facebookresearch / AVT

Code release for ICCV 2021 paper "Anticipative Video Transformer"
Apache License 2.0
152 stars 28 forks source link

EOFError: Ran out of input #45

Open htcboy opened 1 year ago

htcboy commented 1 year ago

Traceback (most recent call last): File "train_net.py", line 44, in main() # pylint: disable=no-value-for-parameter # Uses hydra File "lib/python3.7/site-packages/hydra/main.py", line 95, in decorated_main config_name=config_name, File "lib/python3.7/site-packages/hydra/_internal/utils.py", line 401, in _run_hydra overrides=overrides, File "lib/python3.7/site-packages/hydra/_internal/utils.py", line 458, in _run_app lambda: hydra.run( File "lib/python3.7/site-packages/hydra/_internal/utils.py", line 222, in run_and_report raise ex File "lib/python3.7/site-packages/hydra/_internal/utils.py", line 219, in run_and_report return func() File "lib/python3.7/site-packages/hydra/_internal/utils.py", line 461, in overrides=overrides, File "lib/python3.7/site-packages/hydra/internal/hydra.py", line 132, in run = ret.return_value File "lib/python3.7/site-packages/hydra/core/utils.py", line 260, in return_value raise self._return_value File "lib/python3.7/site-packages/hydra/core/utils.py", line 186, in run_job ret.return_value = task_function(task_cfg) File "train_net.py", line 35, in main getattr(func, cfg.train.fn).main(cfg) File "AVT/func/train.py", line 689, in main init_model(model_to_init, ckpt_path, ckpt_modules_to_keep, logger) File "AVT/func/train.py", line 467, in init_model checkpoint = torch.load(ckpt_path, map_location="cpu") File "lib/python3.7/site-packages/torch/serialization.py", line 795, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "lib/python3.7/site-packages/torch/serialization.py", line 1002, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) EOFError: Ran out of input

i meet this problem when i load the checkpoint and i can't figure it out, so i wonder why this happen? here is the checkpoint: AVT/DATA/pretrained/TIMM/jx_vit_base_p16_224-80ecf9dd.pth