Closed grananqvist closed 2 months ago
Put model to default device before initializing optimizer, because otherwise this would crash for an optimizer like AdaGrad, complaining that optimizer tensors are on cpu, not model tensors are on gpu.
something wrong with poetry.lock...
Put model to default device before initializing optimizer, because otherwise this would crash for an optimizer like AdaGrad, complaining that optimizer tensors are on cpu, not model tensors are on gpu.