SydCaption / SAAT

MIT License
62 stars 21 forks source link

TypeError: __init__() got an unexpected keyword argument 'pretrained' #38

Open bob292 opened 1 year ago

bob292 commented 1 year ago

Dear @SydCaption ,

Thanks for your work! :)

I've been trying to extract 2D feature from a video by using /misc/extract_feats_2D.py, when an error occurs:

Traceback (most recent call last): File "/content/drive/MyDrive/ProCom/SAAT/misc/extract_feats_2D.py", line 64, in extract_feats(opt.file_path, namelist[opt.start_idx:opt.end_idx], opt.frame_per_video, opt.batch_size, save_path) File "/content/drive/MyDrive/ProCom/SAAT/misc/extract_feats_2D.py", line 15, in extract_feats net = resnet101(pretrained=True) File "/content/drive/MyDrive/ProCom/SAAT/misc/resnet.py", line 246, in resnet101 model = ResNet(Bottleneck, [3, 4, 23, 3], **kwargs) TypeError: init() got an unexpected keyword argument 'pretrained'

And it seems to me that ResNet in /3D-ResNets-PyTorch/models/resnet.py doesn't take an argument named 'pretrained'. I therefore wonder if you have any proposition. Thanks in advance!

Yours,