Refactor Trainer arguments, replacing gpus with devices and accelerator

[joelmeyerson]

This commit updates the arguments provided to Trainer in train.py to be compatible with the current PyTorch Lightning version. The gpus keyword argument is now replaced with devices and accelerator.