pytorch-vdsr
pytorch-vdsr copied to clipboard
Training Error
Hi,
Thank you for sharing the code.
I meet the following error whlie training the network. Do you know the reason? Thanks.
#####################
Namespace(batchSize=128, clip=0.4, cuda=True, gpus='0', lr=0.1, momentum=0.9, nEpochs=50, pretrained='', resume='', start_epoch=1, step=10, threads=1, weight_decay=0.0001)
=> use gpu id: '0'
Random Seed: 3131
===> Loading datasets
===> Building model
C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\functional.py:52: UserWarning: size_average and reduce args will be deprecated, please use reduction='sum' instead.
warnings.warn(warning.format(ret))
===> Setting GPU
===> Setting Optimizer
===> Training
Epoch = 1, lr = 0.1
Traceback (most recent call last):
File "main_vdsr.py", line 130, in
Hi, @twtygqyy I solved this problem as suggested https://github.com/twtygqyy/pytorch-vdsr/issues/11 @ZhaoJinHA Thanks.
However, I do not know the reason, do you know why? thanks
for testing, I have the following problem, do you know why? Thanks. @twtygqyy
#####################
=> use gpu id: '0'
C:\ProgramData\Anaconda3\lib\site-packages\torch\serialization.py:425: SourceChangeWarning: source code of class 'torch.nn.modules.container.Sequential' has changed. you can retrieve the original source code by accessing the object's source attribute or set torch.nn.Module.dump_patches = True
and use the patch tool to revert the changes.
warnings.warn(msg, SourceChangeWarning)
Traceback (most recent call last):
File "eval.py", line 33, in
Hi @DanChen001 , the problem is due to the python version. Please refer https://github.com/twtygqyy/pytorch-vdsr/issues/21#issuecomment-372150250 for the solution. And I think the first issue you mentioned is due to training with multiple gpu and testing with single gpu.
Hi @DanChen001 , the problem is due to the python version. Please refer #21 (comment) for the solution. And I think the first issue you mentioned is due to training with multiple gpu and testing with single gpu.
@twtygqyy Thanks. I will try.