RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: train`
What were you doing when this happened? Can you give a full stack trace and the surrounding code?
`Traceback (most recent call last): File "pytorch_connectomics/scripts/main.py", line 72, in <module> main() File "pytorch_connectomics/scripts/main.py", line 65, in main trainer = Trainer(cfg, mode, args.checkpoint, device) File "/n/home11/kguliani/pytorch_connectomics/connectomics/engine/trainer.py", line 29, in _init_ self.model = build_model(self.cfg, self.device) File "/n/home11/kguliani/pytorch_connectomics/connectomics/model/_init_.py", line 27, in build_model model = model.to(device) File "/n/home11/kguliani/.conda/envs/py3_torch/lib/python3.7/site-packages/torch/nn/modules/module.py", line 431, in to device, dtype, non_blocking, convert_to_format = torch._C._nn._parse_to(*args, **kwargs) RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: train`
Is your device “train” by any chance? This is not a device name.