加载自己的模型参数

[Dtappledoghuati]

请问如何在main.py文件中加载自己的模型及参数，我所使用的模型不是pytorch官方提供的，谢谢

[mick.yi]

@Dtappledoghuati 在get_net(net_name, weight_path=None)函数中，把自己的网络构建添加进去就好

[Dtappledoghuati]

谢谢，已成功！

[haobabuhaoba]

谢谢，已成功！

请问你是怎么添加的，能给份代码参考一下吗，我也想试试自己的

[haobabuhaoba]

谢谢，已成功！

[e-mail]
我的邮箱，谢谢你

[nnliuheart]

谢谢，已成功！

[e-mail] 我的邮箱，谢谢你

请问你你解决了吗，我也想知道该怎么做，[e-mail]

[nnliuheart]

谢谢，已成功！
你好，请问你是怎么添加的，我也需要，我也想试试自己的
[e-mail]
我的邮箱，谢谢你