undo change

awslabs · Jan 17, 2023 · c3ac212 · c3ac212
1 parent 73c16fc
commit c3ac212
Showing 1 changed file with 0 additions and 3 deletions.
diff --git a/src/gluonts/torch/model/deep_npts/_estimator.py b/src/gluonts/torch/model/deep_npts/_estimator.py
@@ -350,9 +350,6 @@ def train_model(
             batch_norm=self.batch_norm,
         )
 
-        for parameter in net.parameters():
-            parameter.requires_grad = True
-
         optimizer = torch.optim.Adam(net.parameters(), lr=self.lr)
 
         best_loss = float("inf")