@ptrblck Hello, I have tried anomaly detection and got this error:

RuntimeError                              Traceback (most recent call last)
 in 
18         _,logps = model(images_train)
19         loss = criterion(logps.float(), masks_train.float())
—> 20         loss.backward()
21         optimizer.step()
22         running_loss += loss.item()
~/yes/lib/python3.7/site-packages/torch/tensor.py in backward(self, gradient, retain_graph, create_graph)
193                 products. Defaults to False.
194         “”"
–> 195         torch.autograd.backward(self, gradient, retain_graph, create_graph)
196
197     def register_hook(self, hook):
~/yes/lib/python3.7/site-packages/torch/autograd/init.py in backward(tensors, grad_tensors, retain_graph, create_graph, grad_variables)
97     Variable._execution_engine.run_backward(
98         tensors, grad_tensors, retain_graph, create_graph,
—> 99         allow_unreachable=True)  # allow_unreachable flag
100
101
RuntimeError: Function ‘MseLossBackward’ returned nan values in its 0th output.
Please let me if there is any solution for it.