权重文件

[Afreshbird]

首先感谢您的杰出工作。 当我在在执行 src/test.py 时，在加载权重文件时报错如下：

File "test.py", line 137, in test() File "test.py", line 81, in test Enc, Dec, _ = load(encoder_path, decoder_path) File "test.py", line 44, in load Enc_w = torch.load(encoder_path, map_location=device) File "/home/ws/miniconda3/envs/contrast_test/lib/python3.8/site-packages/torch/serialization.py", line 608, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/home/ws/miniconda3/envs/contrast_test/lib/python3.8/site-packages/torch/serialization.py", line 777, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) _pickle.UnpicklingError: invalid load key, 'v'.

可以确定是是代码写法和权重文件路径完全没问题，所示我想是不是权重文件发生了损坏？

[Afreshbird]

首先感谢您的杰出工作。 当我在在执行 src/test.py 时，在加载权重文件时报错如下：

File "test.py", line 137, in test() File "test.py", line 81, in test Enc, Dec, _ = load(encoder_path, decoder_path) File "test.py", line 44, in load Enc_w = torch.load(encoder_path, map_location=device) File "/home/ws/miniconda3/envs/contrast_test/lib/python3.8/site-packages/torch/serialization.py", line 608, in load return _legacy_load(opened_file, map_location, pickle_module, pickle_load_args) File "/home/ws/miniconda3/envs/contrast_test/lib/python3.8/site-packages/torch/serialization.py", line 777, in _legacy_load magic_number = pickle_module.load(f, pickle_load_args) _pickle.UnpicklingError: invalid load key, 'v'.

可以确定是是代码写法和权重文件路径完全没问题，所示我想是不是权重文件发生了损坏？

不好意思，没点开看，原来需要下载，还以为就是源文件。哈哈哈

[wyhlaowang]

是的，由于权重文件超过github的限制（100MB），git clone 时可能未包含部分权重文件（*.pt），如出现此类情况，可从仓库中单独下载权重文件。