验证已经训练好的Unet模型精度报错

已解决

FDwuyun2023-04-29

1.芯片型号：X3派

2.天工开物开发包OpenExplorer版本：XJ3_OE_2.5.2

3.问题定位：模型精度验证

4.问题具体描述：

利用

python3 tools/train.py --stage float --config configs/segmentation/unet.py

已经训练好Unet模型，接下来想要验证已经训练好的模型精度,运行下面的命令

python3 tools/train.py --stage float --config configs/segmentation/unet.py --val-ckpt float-checkpoint-best.pth.tar --val-only

报错

Traceback (most recent call last):

File "tools/train.py", line 277, in <module>

train(

File "tools/train.py", line 221, in train

setup_args_env(args_env)

File "/home/qlf/.local/lib/python3.8/site-packages/hat/utils/setup_env.py", line 18, in setup_args_env

args_env_infos.update({args_env_info: args_env[i + 1]})

IndexError: list index out of range

算法工具链