Merge branch 'master' into byol_encoder_opt

Lightning-Universe · Jun 16, 2021 · 124fd1b · 124fd1b
2 parents 6d6721f + b087892
commit 124fd1b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pl_bolts/models/regression/logistic_regression.py b/pl_bolts/models/regression/logistic_regression.py
@@ -53,7 +53,7 @@ def training_step(self, batch, batch_idx):
  # flatten any input
  x = x.view(x.size(0), -1)
 
- y_hat = self(x)
+ y_hat = self.linear(x)
 
  # PyTorch cross_entropy function combines log_softmax and nll_loss in single function
  loss = F.cross_entropy(y_hat, y, reduction='sum')