
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 · 247 阅读 · 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 · 613 阅读 · 2 评论