Traceback (most recent call last):


 "train.py", line 314, in <module>

main(args)


 "train.py", line 200, in main

train(segmentation_module, iterator_train, optimizers, history, epoch, args)


 "train.py", line 37, in train

loss, acc = segmentation_module(batch_data)


 "/home/shrek/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/home/shrek/projects/semantic-segmentation-pytorch3/models/models.py", line 33, in forward

(pred, pred_deepsup) = self.decoder(self.encoder(feed_dict['img_data'], return_feature_maps=True))


 "/home/shrek/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/home/shrek/projects/semantic-segmentation-pytorch3/models/models.py", line 264, in forward

x = self.relu1(self.bn1(self.conv1(x)))


 "/home/shrek/.local/lib/python3.5/site-packages/torch/nn/modules/module.py", line 491, in __call__

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


 "/home/shrek/.local/lib/python3.5/site-packages/torch/nn/modules/conv.py", line 301, in forward

self.padding, self.dilation, self.groups)
RuntimeError: Expected object of type torch.FloatTensor but found type torch.cuda.FloatTensor for argument #2 'weight'


"module_stderr": "Traceback (most recent call last):\n

 \"/tmp/ansible_xshVSK/ansible_module_molecule_vagrant.py\", line 417, in <module>\n
main()\n

 \"/tmp/ansible_xshVSK/ansible_module_molecule_vagrant.py\", line 407, in main\n
v = VagrantClient(module)\n

 \"/tmp/ansible_xshVSK/ansible_module_molecule_vagrant.py\", line 287, in __init__\n
self._config = self._get_config()\n

 \"/tmp/ansible_xshVSK/ansible_module_molecule_vagrant.py\", line 333, in _get_config\n
molecule_file = os.environ['MOLECULE_FILE']\n

 \"/home/vfigueiredo/Code/.virtualenvs/ansible-personal/lib/python2.7/UserDict.py\", line 40, in __getitem__\n
raise KeyError(key)\nKeyError: 'MOLECULE_FILE'\n", 

"module_stdout": "", 
