摘要: #1 首先,加载是有条件的,就是两个模型想要加载的参数的名字相同,才能加载进来。 #2 加载的方法有两种,load_state_dict(strict)和update 代码里的方法是对backbone单独做一个Module类,这样不容易出错。 代码中展示了如何把train_net中的backbone 阅读全文
posted @ 2022-05-25 23:16 The1912 阅读(984) 评论(0) 推荐(0) 编辑
摘要: #1 由于之前一直没有用过torch中自带的可视化工具 torchsummary,就安装了试一下 好像用conda无法安装,只能用pip安装。 进入相应的conda虚拟环境输入 pip install torchsummary 注意别忘了关vpn或者代理之类的,不然会报错 #2 起因是这样,安装了后 阅读全文
posted @ 2022-05-25 20:51 The1912 阅读(707) 评论(0) 推荐(0) 编辑