Code fragment metadata

hasIdentifier: https://padme-analytics.de/pht/codeFragment/e2e5b1e66b4906d2c1ba34ac9ee5ecda95fe9f9c746013f06ad9e44b9f9aada5
hasContent: torch.cuda.set_device(args.gpu)

Code fragment is used in the following Trains