0
この質問をrepositoryに投稿してください。MXnet:ファイルからロードされたモデルのトレーニング中にエラーが発生しました
私はチュートリアルhereに従ったファイルに以前に保存されたモデルをロードしようとしています。チュートリアルのように私は、まったく同じコマンドを使用しますが、私は次のエラーメッセージで満たし、次のように
Traceback (most recent call last):
File "test.py", line 153, in <module>
num_epoch=num_epoch)
File "/home/mypath/software/try_mxnet2/mxnet/python/mxnet/module/base_module.py", line 496, in fit
self.update_metric(eval_metric, data_batch.label)
File "/home/mypath/software/try_mxnet2/mxnet/python/mxnet/module/module.py", line 735, in update_metric
self._exec_group.update_metric(eval_metric, labels)
File "/home/mypath/software/try_mxnet2/mxnet/python/mxnet/module/executor_group.py", line 567, in update_metric
for label, axis in zip(labels, self.label_layouts):
TypeError: zip argument #2 must support iteration
ロードのコードとファイルを再訓練は、次のとおりです。
sym, arg_params, aux_params = mx.model.load_checkpoint('../model/test_mymodel', 25)
lenet_model = mx.mod.Module(symbol=sym, context=mx.gpu(), label_names=None)
lenet_model.bind(for_training=True, data_shapes=[('data', (batch_size,3,16,16))],
label_shapes=lenet_model._label_shapes)
lenet_model.set_params(arg_params, aux_params, allow_missing=True)
lenet_model.fit(train_iter,
optimizer='adam',
optimizer_params={'learning_rate':0.001,'wd':0.0005},
eval_metric='acc',
batch_end_callback = mx.callback.Speedometer(batch_size, n_report),
epoch_end_callback = mx.callback.do_checkpoint("../model/test_mymodel", 5),
num_epoch=num_epoch)
私が持っていたよう私が行lenet_model.fit(...)
をコメントアウトすると、エラーは報告されません。読み込まれたモデルを継続的に訓練することができないか、コードに問題があります。
私は親切なソリューションを楽しみにしています。ありがとう!
/model/test_mymodel.pyファイルを共有してもよろしいですか? – rgaut
@ rgaut。 'test_mymodel.py'ファイルはありません。 'test_mymodel'はモデルファイルのプレフィックスです。 'test_mymodel-0000.params'のようなものです。 – pfc
ローカルに再現できる方法はありますか? – rgaut