модель загрузки mxnet sagemaker - PullRequest
Новая
       26

модель загрузки mxnet sagemaker

0 голосов
/ 19 октября 2018

Я пытаюсь загрузить уже обученную модель из sagemaker MXnet.

Однако у меня есть файл model.tar.gz, когда я пытаюсь выполнить 

> %%bash
> tar -xzf model.tar.gz rm model.tar.gz
> prefix = 'model_name' 
> sym, arg_params, aux_params = mx.model.load_checkpoint(prefix, 0) 
> mod = mx.mod.Module(symbol=sym,
> context=ctx, label_names=None) mod.bind(for_training=False, data_shapes=[('data', (1,3,480,480))], label_shapes=mod._label_shapes)
> mod.set_params(arg_params, aux_params)

Iполучаю сообщение об ошибке Ошибка в операторе multibox_target: [09:08:47] src / operator / contrib /./ multibox_target-inl.h: 225: проверка не пройдена: lshape.ndim () == 3 (0 против 3) ярлыкдолжен быть [batch-num_labels - (> = 5)] тензор

Может кто-нибудь помочь мне с этим?

1 Ответ

0 голосов
/ 24 ноября 2018

Я полагаю, что вы должны запустить deploy.py, прежде чем сможете прогнозировать.проверьте инкубатор-mxnet \ example \ ssd \ deploy.py и обратите внимание, что файлы модели должны находиться в подкаталоге каталога, в котором находится deploy.py.

это работало для моей модели на основе resnet50.

python deploy.py --network resnet50 --prefix model2 / model_algo_1 --num-class 2 --data-shape 416

...