RuntimeError: Expected one of cpu, cuda, mkldnn, opengl, opencl, ideep, hip, msnpu device type at start of device string: train`
Hi,
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.
1 Like