nikigoli's picture
Put specifying args.device in main file instead of inside function
9b4dec3 verified