The arcface model cannot be loaded

[alan525]

It's a great project. I follow the instructions to configure the environment. When the code runs to netArc_checkpoint=torch.load(netArc_checkpoint) an error is reported.
..torch/serialization.py line 702, in _legacy_load
result = unpickler.load()
AttributeError: Can't get attribute 'ArcMarginModel' on <module 'models' from 'SimSwap-main/models/init.py'>
How to solve it? Is the model incorrect？

[NNNNAI]

Could you share the command line you used?

[alan525]

command python test_one_image.py --isTrain false --name people --Arc_path arcface_model/arcface_checkpoint.tar --pic_a_path crop_224/6.jpg --pic_b_path crop_224/ds.jpg --output_path output/

I'm sure the download model and path are correct.However, the model loading still failed.
SimSwap/models/fs_model.py line 64
netArc_checkpoint=torch.load(netArc_checkpoint)

[woctezuma]

SimSwap/models/fs_model.py line 64
netArc_checkpoint=torch.load(netArc_checkpoint)

#9, #10, #24