pytorch代码粘贴
Mona-abc
这个作者很懒,什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
工具代码粘贴4——保存日志打印到控制台
先定义一个函数。定义handler 将日志写入文件(file)和控制台(console) import logging def set_logger(log_file_path, logname = None): logger = logging.getLogger() #创建logger logger.setLevel(logging.INFO) #log等级总开关 ...原创 2019-09-12 17:37:21 · 276 阅读 · 0 评论 -
工具代码粘贴3——pytorch载入模型
print('loading checkpoint.......') model_dict = model.state_dict() pretrained_dict = torch.load(weight_path) pretrained_dict = {k:v for k, v in pretrained_dict.items() if k in model_d...原创 2019-07-21 14:54:37 · 656 阅读 · 2 评论
分享