使用argparse模块 进行device-agnostic设备未知时的编码
Posted henry-zhao
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了使用argparse模块 进行device-agnostic设备未知时的编码相关的知识,希望对你有一定的参考价值。
import argparse
import torch
#核心就是argparse类生成参数包parser实例
parser = argparse.ArgumentParser(description=‘PyTorch Example‘)
#通过这个类实例的方法如add_argument添加参数属性,parse_args解析成点运算可访问的参数包形式
parser.add_argument(‘--disable-cuda‘, action=‘store_true‘,
help=‘Disable CUDA‘) #添加属性disable-cuda
args = parser.parse_args()#参数解析,得到一个类似字典的args
args.device = None
if not args.disable_cuda and torch.cuda.is_available():#参数设置要用cuda,且有GPU可用
args.device = torch.device(‘cuda‘)
else:
args.device = torch.device(‘cpu‘)
x = torch.empty((8, 42), device=args.device)
net = Network().to(device=args.device)
以上是关于使用argparse模块 进行device-agnostic设备未知时的编码的主要内容,如果未能解决你的问题,请参考以下文章