I’m trying to display the optimizer currently in use by the model for documentation purposes.
So the following snippet
optimizer = torch.optim.SGD(model.parameters(), lr=1e-5) print(optimizer)
gives the following output
SGD ( Parameter Group 0 dampening: 0 lr: 1e-05 momentum: 0 nesterov: False weight_decay: 0 )
Is there a way in which I could just get
SGD as the output (and same for other optimizers also)?
I figured I could take the substring of the displayed output till the first parenthesis. Is there a cleaner way of doing this?