import os
import datetime
#打印时间
def printbar():
nowtime = datetime.datetime.now().strftime('%Y-%m-%d %H:%M:%S')
print("\n"+"=========="*8 + "%s"%nowtime)
#mac系统上pytorch和matplotlib在jupyter中同时
转载
2024-01-12 18:51:50
63阅读
1.序列化语义保存模型的推荐方法这里主要有两种方法序列化和恢复模型第一种（recommend）只保存和加载模型参数torch.save(the_model.state_dict(), PATH)
the_model = TheModelClass(*args, **kwargs)
the_model.load_state_dict(torch.load(PATH))第二种保存和加载整个模型tor
转载
2023-10-04 19:46:13
64阅读
elif self.cfg.MODEL.ATSS.POSITIVE_TYPE == 'ATSS':
# 注意：以下的过程都是对于一张图片来操作的
num_anchors_per_loc = len(self.cfg.MODEL.ATSS.ASPECT_RATIOS) * self.cfg.MODEL.ATSS.SCALE
转载
2024-07-04 18:18:59
117阅读
