Traceback (most recent call last):


 "launcher.py", line 54, in <module>

path = trainer.train(generator, "/data/unet_trained", training_iters=1406, epochs=100, display_step=100)


 "/test/u-net/ver6/unet.py", line 341, in train

init = self._initialize(training_iters, output_path, restore)


 "/test/u-net/ver6/unet.py", line 298, in _initialize

self.optimizer = self._get_optimizer(training_iters, global_step)


 "/test/u-net/ver6/unet.py", line 281, in _get_optimizer

**self.opt_kwargs).minimize(self.net.cost,
TypeError: __init__() got an unexpected keyword argument 'momentum'

Traceback (most recent call last):


 "/usr/lib/python3.6/threading.py", line 916, in _bootstrap_inner

self.run()


 "/usr/lib/python3.6/threading.py", line 864, in run

self._target(*self._args, **self._kwargs)


 "/usr/local/byteps/launcher/launch.py", line 47, in worker

subprocess.check_call(command, env=my_env, stdout=sys.stdout, stderr=sys.stderr, shell=True)


 "/usr/lib/python3.6/subprocess.py", line 311, in check_call

raise CalledProcessError(retcode, cmd)
subprocess.CalledProcessError: Command 'python3 /usr/local/byteps/tests/test_mxnet.py' returned non-zero exit status 1.
