No checkpoint found in the desired path.

[umang4002]

/data/theory/robustopt/robust_models/cifar_l2_eps_05/checkpoint.pt.best: no checkpoint found.

[umang4002]

RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu this is the error