Put specifying args.device in main file instead of inside function 9b4dec3 verified nikigoli commited on Jul 6