PyTorch ValueError: оптимизатор получил пустой список параметров при построении модели логистической регрессии ⇐ Python

[Anonymous]

Я попробовал создать модель логистической регрессии с помощью nn.Module

Код: Выделить всё

class LogisticRegressionModel(nn.Module):
def __init__(self, input_dim= None) -> None:
super().__init__()
if input_dim is not None:
torch.manual_seed(9)
self.linear = nn.Linear(in_features=input_dim, out_features=1)
else:
self.linear = None
self.input_dim = None

def forward(self, X: Tensor):
if self.linear is None:
input_dim = X.shape[1]
torch.manual_seed(9)
self.linear = nn.Linear(in_features=input_dim, out_features=1)

return torch.sigmoid(self.linear(X))

Учитывая, что при обучении я использую nn.BCEWithLogitsLoss() и optim.SGD(params= model.parameters(), lr= 0.001) в качестве функции потерь и оптимизатора
Теперь, когда я создаю экземпляр и начинаю обучение, я получаю ValueError: оптимизатор получил пустой список параметров

Подробнее здесь: https://stackoverflow.com/questions/798 ... g-a-logist