Can't make grid search example not work

[kostas1515]

Hello and thank you for this repository.
I am trying to run your example train.py following the instructions for the grid search, but I am having trouble.
First it gives me a syntax error:

  File "train.py", line 62
    inputs, targets = inputs.cuda(), targets.cuda(async=True)
                                                  ^
SyntaxError: invalid syntax

When I remove the target definition it gives me a strange value error:

(faster_rcnn) [konsa15@viz01[barkla] scripts]$ CUDA_VISIBLE_DEVICES=0,1 python train.py --config=config.yaml --self_host=2 resnet_cifar
2020-10-13 13:43:44,660 INFO services.py:1164 -- View the Ray dashboard at http://127.0.0.1:8265
Traceback (most recent call last):
File "train.py", line 196, in
skeletor.execute(do_training)
File "/users/konsa15/gridware/share/python/2.7.8/lib/python3.8/site-packages/skeletor/launcher.py", line 237, in execute
_launch_ray_experiments(experiment_fn, args)
File "/users/konsa15/gridware/share/python/2.7.8/lib/python3.8/site-packages/skeletor/launcher.py", line 130, in _launch_ray_experiments
register_trainable('ray_experiment', _real_ray_exp)
File "/users/konsa15/gridware/share/python/2.7.8/lib/python3.8/site-packages/ray/tune/registry.py", line 62, in register_trainable
trainable = wrap_function(trainable, warn=warn)
File "/users/konsa15/gridware/share/python/2.7.8/lib/python3.8/site-packages/ray/tune/function_runner.py", line 477, in wrap_function
raise ValueError(
ValueError: Unknown argument found in the Trainable function. The function args must include a 'config' positional parameter. Any other args must be 'checkpoint_dir'. Found: ['config', 'status_reporter']

Any ideas?