My dataset is Videos so I am running on server which has to use at least 4 gpus.
if args.gpus is not None:
devices = [args.gpus[i] for i in range(args.workers)]
else:
devices = list(range(args.workers))