Ошибка Load_params для сети MXNet

глубокое обучение Python алгоритм
Ошибка Load_params для сети MXNet

При обучении сети MXNet ее обучает серверный графический процессор.Модель:

  • В локальном ЦП на стороне Mac не может вызыватьload_params()модель нагрузки;
  • В GPU на стороне сервера вызов может быть успешным;

Ошибка заключается в следующем:

AssertionError: Parameter conv0_weight is missing in file xxx.params

Почему?

попробуй 1

Ссылаться наОшибки при использовании load_paramsРешение — увеличить префикс сетевой последовательности Sequentialprefixпараметр:

net = nn.HybridSequential(prefix='net_')
with net.name_scope():
    pretrained = vision.get_model('resnet18_v1', pretrained=True).features
    net.add(pretrained)

Однако все равно выдает ошибку...

попробуй 2

Проверьте номер версии MXNet на стороне сервера и на стороне Max:

pip show mxnet
pip show mxnet-cu80

Номер версии действительно другой! Локальный Mac имеет версию 1.1.0, а сервер — версию 1.2.0.

Унифицируйте версии MXNet на обоих концах до версии 1.1.0:

pip install mxnet==1.1.0
pip install mxnet-cu80==1.1.0

Повторно войдите на сервер, обучите модель, экспортируйте модель на локальный Mac и успешно загрузите модель!

Вывод: при установке фреймворка глубокого обучениядолженЧтобы указать версию пакета, убедитесь, что версии локального ЦП и серверного ГП совпадают.

Категории