pytorch学习的一些tips(2)

传参方式:

固定传参:在pycharm窗口中直接写好参数,在脚本中把参数写固定

命令行传参:在terminal中打开虚拟环境,然后输入python 脚本名。

对于argparse的理解:命令行传参的一种方式,提供了一个接口

关于argparse的使用可以分为三步:

1.argparse.ArgumentParser:创建一个ArgumentParser空间

2.parser.add_argument:向空间中添加参数,这些参数有各种各样的属性

3.parser.parse_args():将这个空间中的字符串转化为对象,使python语言可以识别

argparse.ArgumentParser的使用:进行命令行参数的解析,用于频繁修改参数,自动生成帮助文档

class argparse.ArgumentParser(prog=Noneusage=Nonedescription=Noneepilog=Noneparents[]formatter_class=argparse.HelpFormatterprefix_chars='-fromfile_prefix_chars=Noneargument_default=Noneconflict_handler='error'add_help=Trueallow_abbrev=Trueexit_on_error=True)

先创建一个ArgumentParser对象,然后通过parser.add_argument向里添加参数。

parser.add_argument的使用:

第一个参数参数名称形式一般是"--name"
属性nargs有几个参数
type传进来的参数是什么类型的,例如int,float等
default没有具体的值就使用默认的值
action当参数在命令行中出现时,使用的动作的基本类型
help对该参数的解释

opt=parser.parse_args():将参数字符串转化为对象,并将其设为命名空间(ArgumentParser空间)的一个属性。之后可以从opt中获取空间中对应的值。


在gpu上训练

报错:RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu! (when checking argument for argument mat2 in method wrapper_mm)

原因:没有把model移到cuda上

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值