Как исправить код?

Question

[minamisa]

import torch
import torch.nn as nn

class MyModel(nn.Module):
    def __init__(self, inp, out):
        super().__init__()
        self.layers = nn.ModuleList()
        for i in range(10):
            self.layers.add_module(f'layer_{i}',nn.Linear(inp, ?))
            self.layers.add_module(f'act{i}',nn.ReLU())
            inp -= 1
        self.layers.add_module(f'layer_out',nn.Linear(inp, out))

    def forward(self, x):
    for layer in self.layers:
        x = layer(x)
    return x

model = MyModel(inp = 20, out = 10)

какое значение надо указать на месте знака вопроса в "self.layers.add_module(f'layer_{i}',nn.Linear(inp, ?))", чтобы было правильно для схемы на картинке.

Comments

[Vindicar]

Блин, курс-то кто проходит - ты или хабр?
Пиши конкретно, в чём сложность.

[minamisa]

Vindicar,

import torch
import torch.nn as nn

class MyModel(nn.Module):
def init(self, inp, out, hidden_size = 19):
super().init()
self.layers = nn.ModuleList()
for i in range(10):
self.layers.add_module(f'layer_{i}',nn.Linear(inp, hidden_size))
self.layers.add_module(f'act{i}',nn.ReLU())
inp = hidden_size
hidden_size -= 1
self.layers.add_module(f'layer_out',nn.Linear(inp, out))

def forward(self, x):
    outputs = []
    for i, layer in enumerate(self.layers):
        x = layer(x)
    outputs.append(x)
    return outputs
model = MyModel(inp = 20, out = 10)

Вот код с ModuleList, а ошибка "Метод __init__ должен содержать три аргумента: self, размер входного и выходного тензора."

import torch
import torch.nn as nn

class MyModel(nn.Module):
    def __init__(self, inp, out, hidden_size=19):
        super().__init__()
        layers = []
        for i in range(1, 10):
            layers.append(nn.Linear(inp, hidden_size))
            layers.append(nn.ReLU())
            inp = hidden_size
            hidden_size -= 1
        layers.append(nn.Linear(inp, out))
        self.model = nn.Sequential(*layers)
    def forward(self, x):
        out self.model(x)
        return out

model = MyModel(inp=20, out=10)

а это правильный код только без ModuleList, а надо только с ним

[Vindicar]

minamisa, что-то у тебя первый код кривой, как не знаю что.
Отступы куда-то подевались, подчёркивания вокруг названия __init__() куда-то подевались, в цикле создаёшь слои одного и того же размера, а их размер должен уменьшаться. Вполне очевидно, что первый и последний слой должны быть вне цикла, так как их размеры не вписываются в логику "19, 18, ..., 11". В общем, исправляй сначала это.

Answer 1

[Ланской Кирилл]

Твои вопросы удаляют, потому что ты спрашиваешь задачи, а это нельзя правилами этого сайта, поэтому хотя бы заголовок должен содержать в себе не как исправить код, а как реализовать количество входов и выходов…, короче что-то более обобщённое, что не зависит от конкретной задачи. А здесь кол-во входов этого слоя равно кол-ву предыдущего.

И для лучшего понимания нейросетей вообще советую прочитать эту и ту статью, про которую там говориться: https://habr.com/ru/articles/799123/(это моя статья и я уже несколько раз проверял успешно на её понятность).