Type erorr about torch.FloatTensor and torch.cuda.FloatTensor

[zhouhongh]

Hi:
I followed your instructions in readme,but I got the following error:
Traceback (most recent call last):
File "test_refine_w_mask_two_path.py", line 149, in
a.test()
File "test_refine_w_mask_two_path.py", line 124, in test
y_pred_before_refine, y_pred, mu, logvar, flow, flowback, mask_fw, mask_bw = vae(frame1, data, bg_mask, fg_mask, noise_bg, z_m)
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/zhouhh/seg2vid/src/models/multiframe_w_mask_genmask_two_path.py", line 254, in forward
enco1, enco2, enco3, codex = self.encoder(input).cuda()
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/zhouhh/seg2vid/src/models/multiframe_w_mask_genmask_two_path.py", line 157, in forward
enco1 = self.econv1(x) # 32
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/zhouhh/seg2vid/src/models/multiframe_w_mask_genmask_two_path.py", line 91, in forward
return self.main(x)
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/container.py", line 91, in forward
input = module(input)
File "/home/zhouhh/anaconda3/envs/python35/lib/python3.5/site-packages/torch/nn/modules/module.py", line 477, in call
result = self.forward(*input, **kwargs)
File "/home/zhouhh/anaconda3/envs/python35/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'
Could you give me some advices?