Traceback (most recent call last):


 "train_ssd.py", line 142, in <module>

main()


 "train_ssd.py", line 133, in main

model = train(cfg, args)


 "train_ssd.py", line 34, in train

model.to(device)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\nn\modules\module.py", line 386, in to

return self._apply(convert)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\nn\modules\module.py", line 193, in _apply

module._apply(fn)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\nn\modules\module.py", line 193, in _apply

module._apply(fn)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\nn\modules\module.py", line 199, in _apply

param.data = fn(param.data)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\nn\modules\module.py", line 384, in convert

return t.to(device, dtype if t.is_floating_point() else None, non_blocking)


 "D:\Anaconda3\envs\SSD\lib\site-packages\torch\cuda\__init__.py", line 163, in _lazy_init

torch._C._cuda_init()
RuntimeError: CUDA error: unknown error
Traceback (most recent call last):


 "/content/OpenNMT-py/onmt/trainer.py", line 347, in _gradient_accumulation

trunc_size=trunc_size)


 "/content/OpenNMT-py/onmt/utils/loss.py", line 158, in __call__

loss, stats = self._compute_loss(batch, **shard)


 "/content/OpenNMT-py/onmt/utils/loss.py", line 233, in _compute_loss

scores = self.generator(bottled_output)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 489, in __call__

result = self.forward(*input, **kwargs)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/container.py", line 92, in forward

input = module(input)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/module.py", line 489, in __call__

result = self.forward(*input, **kwargs)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/modules/linear.py", line 67, in forward

return F.linear(input, self.weight, self.bias)


 "/usr/local/lib/python3.6/dist-packages/torch/nn/functional.py", line 1352, in linear

ret = torch.addmm(torch.jit._unwrap_optional(bias), input, weight.t())
RuntimeError: Expected object of backend CPU but got backend CUDA for argument #4 'mat1'
