search

hyf015 / egocentric-gaze-prediction

Code for the paper "Predicting Gaze in Egocentric Video by Learning Task-dependent Attention Transition"
62 stars 18 forks source link

late_fusion.py does not work #1

Closed kazucmpt closed 5 years ago

kazucmpt commented 5 years ago

late_fusion.py does not work because you cannot cat two tensors whose shapes are different from each other. The dimensions of f and g are (10,1,14,14) and (10,1,224,224) in line 20.

I think you should not comment out in line 19. You have to change the dimension of f from (10,1,14,14) to (10,1,224,224) by upsampling.

hyf015 commented 5 years ago

Yes, it is commented. The code has been updated, maybe you are using a version before update.

kazucmpt commented 5 years ago

Thank you very much.