2017-09-25 4 views
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(...)をコメントアウトすると、エラーは報告されません。読み込まれたモデルを継続的に訓練することができないか、コードに問題があります。

私は親切なソリューションを楽しみにしています。ありがとう!

+0

/model/test_mymodel.pyファイルを共有してもよろしいですか? – rgaut

+0

@ rgaut。 'test_mymodel.py'ファイルはありません。 'test_mymodel'はモデルファイルのプレフィックスです。 'test_mymodel-0000.params'のようなものです。 – pfc

+0

ローカルに再現できる方法はありますか? – rgaut

答えて

関連する問題