I got the error while runing python main.py, could you please advise.

[furqan4545]

Error log:

(FuSta_CVPR2020) azureuser@transpifytts:~/FuSta/CVPR2020CODE_yulunliu_modified$ python main.py ../input_frames/ ../CVPR2020_output/ ../CVPR2020_warping_field/ Loading weights for net_encoder Loading weights for net_decoder loaded!!!!!!!!!!!!!!!!! ../input_frames/ 1.0 0.99999994 1.0000001 THCudaCheck FAIL file=/pytorch/aten/src/THC/THCGeneral.cpp line=405 error=8 : invalid device function Traceback (most recent call last): File "main.py", line 583, in optic, video, mask, mask_object = compute_flow_seg(video, H, ite2 (SEG - 1)) File "main.py", line 219, in compute_flow_seg ss, ss_person = compute_mask(video[i, :, :, :]) File "main.py", line 123, in compute_mask pred_tmp = segmentation_module(feed_dict, segSize=segSize) File "/home/azureuser/miniconda3/envs/FuSta_CVPR2020/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(input, kwargs) File "/home/azureuser/FuSta/CVPR2020CODE_yulunliu_modified/models/models.py", line 46, in forward pred = self.decoder(self.encoder(feed_dict['img_data'], return_feature_maps=True), segSize=segSize) File "/home/azureuser/miniconda3/envs/FuSta_CVPR2020/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, *kwargs) File "/home/azureuser/FuSta/CVPR2020CODE_yulunliu_modified/models/models.py", line 254, in forward x = self.relu1(self.bn1(self.conv1(x))) File "/home/azureuser/miniconda3/envs/FuSta_CVPR2020/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(input, kwargs) File "/home/azureuser/miniconda3/envs/FuSta_CVPR2020/lib/python3.6/site-packages/torch/nn/modules/activation.py", line 50, in forward return F.threshold(input, self.threshold, self.value, self.inplace) File "/home/azureuser/miniconda3/envs/FuSta_CVPR2020/lib/python3.6/site-packages/torch/nn/functional.py", line 838, in threshold result = VF.threshold(input, threshold, value) RuntimeError: CUDA error: no kernel image is available for execution on the device

###########################

Dear author, could you please advise on this error.

[caibucai22]

@furqan4545 I also got this problem, have you found solutions?