Grad-CAM.pytorch
Grad-CAM.pytorch copied to clipboard
加载自己的模型参数
请问如何在main.py文件中加载自己的模型及参数,我所使用的模型不是pytorch官方提供的,谢谢
@Dtappledoghuati 在get_net(net_name, weight_path=None)
函数中,把自己的网络构建添加进去就好
谢谢,已成功!
谢谢,已成功!
请问你是怎么添加的,能给份代码参考一下吗,我也想试试自己的
谢谢,已成功! 你好,请问你是怎么添加的,我也需要,我也想试试自己的 [email protected] 我的邮箱,谢谢你