search

wilson1yan / VideoGPT

MIT License
962 stars 115 forks source link

i3d loading error #44

Open rook86 opened 2 months ago

rook86 commented 2 months ago

I got the following error when loading i3d. Does anyone know how to solve this?

File "/home/taiju/pyen/VDT/common_metrics_on_video_quality/calculate_fvd.py", line 30, in calculate_fvd
    i3d = load_i3d_pretrained(device=device)
  File "/home/taiju/pyen/VDT/common_metrics_on_video_quality/fvd/videogpt/fvd.py", line 26, in load_i3d_pretrained
    i3d.load_state_dict(torch.load(filepath, map_location=device))
  File "/home/taiju/pyen/VDT/env/lib/python3.8/site-packages/torch/serialization.py", line 712, in load
    return _load(opened_zipfile, map_location, pickle_module, **pickle_load_args)
  File "/home/taiju/pyen/VDT/env/lib/python3.8/site-packages/torch/serialization.py", line 1049, in _load
    result = unpickler.load()
  File "/home/taiju/pyen/VDT/env/lib/python3.8/site-packages/torch/serialization.py", line 1019, in persistent_load
    load_tensor(dtype, nbytes, key, _maybe_decode_ascii(location))
  File "/home/taiju/pyen/VDT/env/lib/python3.8/site-packages/torch/serialization.py", line 1001, in load_tensor
    wrap_storage=restore_location(storage, location),
  File "/home/taiju/pyen/VDT/env/lib/python3.8/site-packages/torch/serialization.py", line 976, in restore_location
    result = map_location(storage, location)
TypeError: 'int' object is not callable