Python — od podstaw do zaawansowanego · PyTorch — tensory, autograd, prosty MLP
nn.Module — definicja modelu
PyTorch — tensory, autograd, prosty MLP
Wprowadzenie
nn.Module to bazowa klasa dla modeli. Dziedziczysz, definiujesz warstwy w __init__, forward pass w forward(). __call__ deleguje do forward + hooks. Parameter discovery automatyczny.