./train_atari.sh is running but need two changes in code.

[tshh]

have to modify code in two places:
in model_loader.py :
#model.cuda(device_id=args.gpu)
model.cuda(args.gpu)
and model_interface.py
#self.models[key].cuda(device_id=gpu_id)
self.models[key].cuda(gpu_id)
because i got :
TypeError: cuda() got an unexpected keyword argument 'device_id'

./train_atari.sh is running and gpu is beasy.

[EliasVansteenkiste]

device_id keyword changed to device, so you can also use .cuda(device=gpu_id)