![](https://img-blog.csdnimg.cn/20201014180756922.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch代码粘贴
Mona-abc
这个作者很懒,什么都没留下…
展开
-
工具代码粘贴4——保存日志打印到控制台
先定义一个函数。定义handler 将日志写入文件(file)和控制台(console)import loggingdef set_logger(log_file_path, logname = None): logger = logging.getLogger() #创建logger logger.setLevel(logging.INFO) #log等级总开关...原创 2019-09-12 17:37:21 · 212 阅读 · 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 · 571 阅读 · 2 评论