mxnet加载模型 IndexError: list index out of range

程序员文章站 2022-05-27 10:32:53

...

loading ['/data/user1/log/frvt/models/r34_webface_mask/siamese/siamese34-arcface-webfaceSiamese/model-0009.params']
Traceback (most recent call last):
  File "train_0305.py", line 447, in <module>
    main()
  File "train_0305.py", line 443, in main
    train_net()
  File "train_0305.py", line 226, in train_net
    _, arg_params, aux_params = mx.model.load_checkpoint(vec[0], int(vec[1]))
IndexError: list index out of range

不要慌，你加载的格式写错了，前缀和epoch要逗号分开
错误写法：

default.pretrained = '/data/user1/log/frvt/models/r34_webface_mask/siamese/siamese34-arcface-webfaceSiamese/model-0009.params'

正确写法：

default.pretrained = '/data/user1/log/frvt/models/r34_webface_mask/siamese/siamese34-arcface-webfaceSiamese/model,9'

参考