06 2021 档案

torch.load()的作用

摘要：torch.load()的作用：从文件加载用torch.save()保存的对象。 api： torch.load(f, map_location=None, pickle_module=<module 'pickle' from '/opt/conda/lib/python3.6/pickle.py 阅读全文

posted @ 2021-06-10 16:57 鲍曼小学生阅读(4763) 评论(0) 推荐(0) 编辑

遇到问题：pytorch结果不可重现

摘要：pytorch结果不可重现，大概率是随机数种子设置的问题。 Pytorch、Python、Numpy中的随机种子没有固定数据预处理、增强方式采用了概率，若没有设置固定的随机种子，结果可能不同。例如常用数据增强库albumentations就采用了Python的随机产生器。训练数据集被随机打乱了顺阅读全文

posted @ 2021-06-10 16:28 鲍曼小学生阅读(326) 评论(0) 推荐(0) 编辑

NNCF框架自用流程

摘要：阅读全文

posted @ 2021-06-08 17:10 鲍曼小学生阅读(408) 评论(0) 推荐(0) 编辑

Python中__all__的作用

摘要：__all__ = [<string>] 它是一个string元素组成的list变量，定义了当你使用 from <module> import * 导入某个模块的时候能导出的符号（这里代表变量，函数，类等）。其实就是代码保护，限定本模块中只有哪些能被import。举例：foo.py __all_ 阅读全文

posted @ 2021-06-03 19:30 鲍曼小学生阅读(133) 评论(0) 推荐(0) 编辑

pytorch--nn.module里的parameters和buffer的区别与定义

摘要：定义：在模型中有两种需要被保存下来的参数：parameter和buffer 一种是反向传播需要被optimizer更新的，称之为 parameter（如权重等）一种是反向传播不需要被optimizer更新，称之为 buffer（一些阈值之类的）注册：torch.nn.register_para 阅读全文

posted @ 2021-06-03 10:43 鲍曼小学生阅读(1707) 评论(0) 推荐(0) 编辑