TRN-pytorch
TRN-pytorch copied to clipboard
test_video.py - AttributeError: 'NoneType' object has no attribute 'groups'
Hi,
I tried to run on the pre-trained example and got the following error I am running on Windows 10, Anaconda, Python 3.6, Pytorch 0.3.0 etc, any idea why?:
(momentsintime) C:\Users\Pablo\TRN-pytorch>python test_video.py --arch InceptionV3 --dataset moments --weight pretrain\TRN_moments_RGB_InceptionV3_TRNmultiscale_segment8_best.pth.tar --video_file 'C:\Users\Pablo\TRN-pytorch\sample_data\bolei_juggling.mp4' --rendered_output 'C:\Users\Pablo\TRN-pytorch\sample_data\predicted_video.mp4'
Multi-Scale Temporal Relation Network Module in use ['8-frame relation', '7-frame relation', '6-frame relation', '5-frame relation', '4-frame relation', '3-frame relation', '2-frame relation']
Freezing BatchNorm2D except the first one.
Extracting frames using ffmpeg...
Traceback (most recent call last):
File "test_video.py", line 133, in
ubuntu18 have the same AttributeError,too.
I run the same program successed , my computer is ubuntu18 + cuda 9.0 + cudnn 7.4.2 + python3.6 + torch 1.0
Has this problem been solved?