salmank255 / ROAD_Waymo_Baseline

Other
15 stars 0 forks source link

slowfast's weight can't load #15

Open ding1007 opened 1 month ago

ding1007 commented 1 month ago

i have downloaded weight SLOWFAST_R101_K700.pth.tar ,when i choose model_type slowfast , the error is : File "/home/ubuntu/work/ROAD_Waymo_Baseline/utils.py", line 179, in load_pretrain checkpoint = torch.load(pretrain_opt.path, map_location=lambda storage, loc: storage.cuda()) File "/home/ubuntu/mambaforge/envs/torch/lib/python3.10/site-packages/torch/serialization.py", line 815, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/home/ubuntu/mambaforge/envs/torch/lib/python3.10/site-packages/torch/serialization.py", line 1033, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) _pickle.UnpicklingError: invalid load key, '<'.

salmank255 commented 1 month ago

Hi @ding1007,

Your weights are not downloaded correctly. I think it's best to check the file size of your weights in "kinetics-pt" directory.