I have a program as in the attached file. The error is
Loss is nan, stopping training {\'loss_classifier\': tensor(nan, device=\'cuda:0\', grad_fn=