search

gsig / charades-algorithms

Activity Recognition Algorithms for the Charades Dataset
http://allenai.org/plato/charades/
201 stars 96 forks source link

[question]Loading Pre-trained flow model. #13

Closed ksammoh closed 5 years ago

ksammoh commented 5 years ago

Hi,

I came across your data set and repo while researching about video analysis. Thank you very much for sharing your rich data set and code.

I'm using pytorch implementation and able to load the rgb model without any problem. But I', facing problems while loading flow model downloaded from https://www.dropbox.com/s/m1hkeiwjtndt26z/twostream_flow.pth?dl=1.

Specifically, I face error @ # replace last layer block of code in __init__.py file in the final else case. System says "Dropout does not have field in_features..."

Could you please give me some direction on how to overcome this issue? Thanks in advance!

gsig commented 5 years ago

Seems like I introduced code that didn't support the flow model at some point. Should be fixed now in the latest commit. Let me know if it works.

ksammoh commented 5 years ago

Thank you very much Gunnar. I tested your changes and they are working fine.

Thanks a lot for quick reply!