代码位置:main.py文件中的main_worker()函数(修改部分为字体加粗部分): elif args.gpu is not None: torch.cuda.set_device(args.gpu) model = model.
確定! 回上一頁