yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021
MIT License
187 stars 34 forks source link

Global output shape different for train and eval #53

Open Saurabh1409 opened 1 year ago

Saurabh1409 commented 1 year ago

Why is the shape of global output different during training and evaluation for the same input video? I have passed the same vid input to train.py and test.py, but the global output shape differs for both of these cases. _vid_id : 01KML global_output (train.py): torch.Size([38, 1936]) globaloutput (test.py): torch.Size([272, 1936]) I am unable to understand this requirement. Can you please help me with this?