Great that works for my model, but there are one or two other variables that I need to convert (that are outside my model).
I’ve tried applying this to my optimizer, and it yields AttributeError: 'Adam' object has no attribute 'double' (when calling optim.double())· I’d assume this is because model subclasses from nn.Module whereas optim subclasses from torch.optim.Optimizer. Is there a way to apply this to the optimzier as well?
Thank you!