Colecții Dinamice de Straturi: nn.ModuleList și nn.ModuleDict în PyTorch & Deep Learning Practic

import torch
import torch.nn as nn

class DeepMLP(nn.Module):
    def __init__(self, in_features, hidden_size, num_layers):
        super().__init__()
        # Înregistrare corectă a listei de straturi
        self.layers = nn.ModuleList([
            nn.Linear(in_features if i == 0 else hidden_size, hidden_size)
            for i in range(num_layers)
        ])
        self.activation = nn.GELU()

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

model = DeepMLP(64, 128, 4)
print("Număr parametri înregistrați:", len(list(model.parameters())))