Bohdan Pytaychuk

@mostoo45 Restarting the kernel with clearing an output worked for me

Hi @VainF , got the same issue. Do you know what it can be related to?

@kinfeparty @VainF Found the issue. According to [Pytorch optimizer documentation](https://pytorch.org/docs/stable/optim.html), > if you need to move a model to GPU via .cuda(), please do so before constructing optimizers for it....