I am running this example code on CPU on baremetal machine:
I get the error:
Traceback (most recent call last):
File “ray_sgd_example_code.py”, line 146, in
train_example(num_workers=args.num_workers, use_gpu=args.use_gpu)
File “ray_sgd_example_code.py”, line 97, in train_example
m = trainer1.get_model(to_cpu=ray.util.client.ray.is_connected())
TypeError: get_model() got an unexpected keyword argument ‘to_cpu’