摘要: from thop import profile class Test(nn.Module): def __init__(self, input_size, output_szie): super(Test, self).__init__() self.out = nn.Linear(input_s 阅读全文
posted @ 2020-10-25 20:43 douzujun 阅读(619) 评论(0) 推荐(0) 编辑
摘要: 笔记 import logging def create_logger(fp): #打印日志的时间、日志级别名称、日志信息 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s: %(message)s 阅读全文
posted @ 2020-10-25 20:05 douzujun 阅读(166) 评论(0) 推荐(0) 编辑