MrBurnst opened this issue 6 years ago · 1 comments
it should be
if __name__ == "__main__": worker = PyTorchWorker(run_id='0')
instead of
if __name__ == "__main__": worker = KerasWorker(run_id='0')
in the example.
Thanks for pointing that out! The good old copy-paste error :)