search

linjieli222 / HERO

Research code for EMNLP 2020 paper "HERO: Hierarchical Encoder for Video+Language Omni-representation Pre-training"
https://arxiv.org/abs/2005.00200
MIT License
230 stars 34 forks source link

I found a bug in pretrain.py #31

Closed liveseongho closed 3 years ago

liveseongho commented 3 years ago

Hi @linjieli222 https://github.com/linjieli222/HERO/blob/792808562e9f981656e756753f577ab0e5be738b/pretrain.py#L41

This line makes on error as follows: ModuleNotFoundError: No module named '__main__.load_data'; '__main__' is not a package

This line should be modified to from load_data import load_video_sub_dataset