pytorch
大石头2409
这个作者很懒,什么都没留下…
展开
-
pytorch保存模型等相关参数,利用torch.save(),以及读取保存之后的文件
一、如何保存模型参数,优化器参数 假设在某个epoch,我们要保存模型参数,优化器参数以及epoch。1. 先建立一个字典,保存三个参数:if steps % 10000 == 0: file_path = os.path.join(args.weight_dir, 'weights-{:07d}.pth.tar'.format(steps)) ...原创 2018-12-06 22:34:20 · 4848 阅读 · 0 评论 -
argparse python
1、使用步骤:(1)import argparse 首先导入模块(2)parser = argparse.ArgumentParser() 创建一个解析对象(3)parser.add_argument() 向该对象中添加你要关注的命令行参数和选项(4)parser.parse_args() 进行解析2、argparse.ArgumentParser(...原创 2018-12-07 10:57:44 · 194 阅读 · 0 评论