automl/HpBandSter

small error in example 5 - Pytorch

MrBurnst opened this issue · 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 :)