`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`