pytorch自己构建网络将变量添加到Parameter中

最新推荐文章于 2024-07-02 10:24:12 发布

weixin_45716266

最新推荐文章于 2024-07-02 10:24:12 发布

阅读量919

点赞数

文章标签： python pytorch 深度学习人工智能

本文链接：https://blog.csdn.net/weixin_45716266/article/details/122612054

版权

本文介绍了在PyTorch中构建网络时，如何正确将变量添加到Parameter以便于优化。错误的用法会导致optimizer无法识别参数，而正确的写法能确保参数被正确管理。

摘要由CSDN通过智能技术生成

错误用法：

class Ant_Model(nn.Module):
    def __init__(self, alpha, J, K, batch_size):
        super(Ant_Model, self).__init__()
        self.J = J
        self.K = K
        self.alpha = alpha
        self.batch_size = batch_size
        self.dual_1 = torch.ones((self.K, 1

最低0.47元/天解锁文章

确定要放弃本次机会？

福利倒计时

: :

立减 ¥

普通VIP年卡可用

立即使用

weixin_45716266

关注关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
pytorch自己构建网络将变量添加到Parameter中

错误用法：class Ant_Model(nn.Module): def __init__(self, alpha, J, K, batch_size): super(Ant_Model, self).__init__() self.J = J self.K = K self.alpha = alpha self.batch_size = batch_size self.dual_1 = torch.o
复制链接

扫一扫