Merge pull request #116 from GoBeromsu/master

Author: kulbachcedric

deep_river/anomaly/ae.py

@@ -180,9 +180,9 @@ def _learn(self, x: torch.Tensor) -> "Autoencoder":
         self.module.train()
         x_pred = self.module(x)
         loss = self.loss_func(x_pred, x)
+        self.optimizer.zero_grad()
         loss.backward()
         self.optimizer.step()
-        self.optimizer.zero_grad()
         return self
 
     def score_one(self, x: dict) -> float: