cavalleria / cavaface

face recognition training project(pytorch)
MIT License
456 stars 88 forks source link

Model loading error #71

Closed Youngboy12 closed 3 years ago

Youngboy12 commented 3 years ago

When I run bash run.sh for evaluation,there is a error happened: Traceback (most recent call last): File "./main.py", line 179, in _main(args) File "./main.py", line 68, in _main net = _load_model(args) File "./main.py", line 31, in _load_model net = get_infer(args) File "/remote_projects/cavaface/evaluation/infer/init.py", line 8, in get_infer return CitrusPytorchInfer(args, dtype=dtype) File "/remote_projects/cavaface/evaluation/infer/citrus_pytorch_infer.py", line 20, in init super(CitrusPytorchInfer, self).init(args=args, dtype=dtype) File "/remote_projects/cavaface/evaluation/infer/citrus_base_infer.py", line 260, in init self._load_model() File "/remote_projects/cavaface/evaluation/infer/citrus_pytorch_infer.py", line 37, in _load_model tnet.model=torch.jit.load(spath[0], map_location=tnet.ctx) File "/root/anaconda3/lib/python3.7/site-packages/torch/jit/init.py", line 221, in load raise ValueError("The provided filename {} does not exist".format(f)) ValueError: The provided filename /workspace/results/lyb_results/IR_SE_100_Arcface/model_ep24_1.pth does not exist

Can you help me about that?

cavalleria commented 3 years ago

please set model path properly

Youngboy12 commented 3 years ago

I got it, Thanks.