ysh329 commented on a change in pull request #7363: Add tensorboard configure into ./common/fit.py and ./train_mnist.py URL: https://github.com/apache/incubator-mxnet/pull/7363#discussion_r131638833
########## File path: example/image-classification/train_mnist.py ########## @@ -75,5 +75,13 @@ def get_mnist_iter(args, kv): net = import_module('symbols.'+args.network) sym = net.get_symbol(**vars(args)) + # tensorboard logs + train_log = 'logs/mnist/train' + eval_log = 'logs/mnist/eval' + batch_end_callbacks = [mx.contrib.tensorboard.LogMetricsCallback(train_log)] Review comment: @zihaolucky After `pip install tensorboard` in Docker, I found another problem :sob: ```Shell root@65ea267b5b52:~/incubator-mxnet-0.10.0.post2/example/image-classification# python train_mnist.py INFO:root:start with arguments Namespace(add_stn=False, batch_size=64, disp_batches=100, dtype='float32', gpus=None, kv_store='device', load_epoch=None, lr=0.05, lr_factor=0.1, lr_step_epochs='10', model_prefix=None, mom=0.9, monitor=0, network='mlp', num_classes=10, num_epochs=20, num_examples=60000, num_layers=None, optimizer='sgd', test_io=0, top_k=0, wd=0.0001) Traceback (most recent call last): File "train_mnist.py", line 87, in <module> eval_end_callback = eval_end_callbacks) File "/root/incubator-mxnet-0.10.0.post2/example/image-classification/common/fit.py", line 197, in fit monitor = monitor) File "/mxnet/python/mxnet/module/base_module.py", line 465, in fit optimizer_params=optimizer_params) File "/mxnet/python/mxnet/module/module.py", line 478, in init_optimizer **optimizer_params) File "/mxnet/python/mxnet/optimizer.py", line 128, in create_optimizer return Optimizer.opt_registry[name.lower()](**kwargs) File "/mxnet/python/mxnet/optimizer.py", line 328, in __init__ super(SGD, self).__init__(**kwargs) TypeError: __init__() got an unexpected keyword argument 'multi_precision' ``` ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services