我的带有可变隐藏层的线性模型看起来还不错吗? [Pytorch]

发布于 2025-02-13 07:50:46 字数 634 浏览 1 评论 0原文

我只需要一些帮助,因为我获得了很高的损失价值,这是我创建的第一个模型,因此不确定它是原因是我的模型或输入数据。

我的代码在下面

class Net(nn.Module):

    def __init__(self, D_in, H, D_out,num_layers,dropout):
        super().__init__()
        self.fcs = nn.ModuleList([nn.Linear(D_in, H)])
        for k in range(num_layers-1):
            self.fcs.append(nn.Linear(H, H))
            
        self.fcs.append(nn.Linear(H, D_out))
        self.relu = nn.ReLU()
        self.dropout = nn.Dropout(dropout)

    def forward(self, x):
      
        for l in self.fcs:
            x = self.dropout(x) 
            x = self.relu(l(x))

        return x.squeeze()

I just need some help as I am getting very high loss values and this is the first model ive created myself so not sure if its my model or input data that is the cause.

My code is below

class Net(nn.Module):

    def __init__(self, D_in, H, D_out,num_layers,dropout):
        super().__init__()
        self.fcs = nn.ModuleList([nn.Linear(D_in, H)])
        for k in range(num_layers-1):
            self.fcs.append(nn.Linear(H, H))
            
        self.fcs.append(nn.Linear(H, D_out))
        self.relu = nn.ReLU()
        self.dropout = nn.Dropout(dropout)

    def forward(self, x):
      
        for l in self.fcs:
            x = self.dropout(x) 
            x = self.relu(l(x))

        return x.squeeze()

如果你对这篇内容有疑问,欢迎到本站社区发帖提问 参与讨论,获取更多帮助,或者扫码二维码加入 Web 技术交流群。

扫码二维码加入Web技术交流群

发布评论

需要 登录 才能够评论, 你可以免费 注册 一个本站的账号。
列表为空,暂无数据
我们使用 Cookies 和其他技术来定制您的体验包括您的登录状态等。通过阅读我们的 隐私政策 了解更多相关信息。 单击 接受 或继续使用网站,即表示您同意使用 Cookies 和您的相关数据。
原文