Failure of training with CPU mode

[tqyunwuxin]

Evaluation mode Checkpoints from ../results_gaussian_denoising/pretrained_sigma70 Writing to ../results_gaussian_denoising/pretrained_sigma70/test04/_data.pkl Epoch 51 Testing on Set12 Traceback (most recent call last): File "main.py", line 281, in main() File "main.py", line 257, in main run(my_experiment) File "main.py", line 239, in run evaluate(experiment) File "main.py", line 160, in evaluate test_epoch(epoch, experiment) File "main.py", line 119, in test_epoch pred = net(inputs) File "/home/tuqiang/anaconda3/envs/pt_py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, **kwargs) File "/media/tuqiang/16af747f-50d9-46bf-82c1-788861185451/workspace/denoise_new/deeplearning_denoise/n3net/src_denoising/models/n3net.py", line 231, in forward x = self.nlsi File "/home/tuqiang/anaconda3/envs/pt_py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, **kwargs) File "/media/tuqiang/16af747f-50d9-46bf-82c1-788861185451/workspace/denoise_new/deeplearning_denoise/n3net/src_denoising/models/n3net.py", line 184, in forward x = self.n3aggregation(xg,xe,ye,log_temp=log_temp) File "/home/tuqiang/anaconda3/envs/pt_py36/lib/python3.6/site-packages/torch/nn/modules/module.py", line 489, in call result = self.forward(*input, **kwargs) File "/media/tuqiang/16af747f-50d9-46bf-82c1-788861185451/workspace/denoise_new/deeplearning_denoise/n3net/src_denoising/models/non_local.py", line 274, in forward x_patch, padding = ops.im2patch(x, self.patchsize, self.stride, None, returnpadding=True) File "/media/tuqiang/16af747f-50d9-46bf-82c1-788861185451/workspace/denoise_new/deeplearning_denoise/n3net/src_denoising/ops.py", line 121, in im2patch x2col = pyinn.im2col(xpad, [patchsize]*2, [stride]*2, [0,0]) NameError: name 'pyinn' is not defined

[tobiasploetz]

Hi @tqyunwuxin,

currently we only support GPU computation for N3Aggregation2D since im2col currently has no CPU implementation. It's probably not that hard to use some numpy functions to implement im2col on CPU, though.

Bests, Tobias