unet网络测试遇到的报错问题及解决

1、出现 No such file or directory: 'checkpoint_epoch5.pth'

Traceback (most recent call last):
  File "predict.py", line 91, in 
    net.load_state_dict(torch.load(args.model, map_location=device))
  File "D:\ProgramData\Miniconda3\lib\site-packages\torch\serialization.py", line 699, in load
    with _open_file_like(f, 'rb') as opened_file:
  File "D:\ProgramData\Miniconda3\lib\site-packages\torch\serialization.py", line 231, in _open_file_like
    return _open_file(name_or_buffer, mode)
  File "D:\ProgramData\Miniconda3\lib\site-packages\torch\serialization.py", line 212, in __init__
    super(_open_file, self).__init__(open(name, mode))
FileNotFoundError: [Errno 2] No such file or directory: 'checkpoint_epoch5.pth'

 解决办法： 将训练生成的模型文件checkpoint_epoch5.pth放在与predict.py同级的目录下，如图：