update

paartheee · Mar 6, 2024 · 0a62be3 · 0a62be3
1 parent c413222
commit 0a62be3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/diffmot.py b/diffmot.py
@@ -62,7 +62,7 @@ def train(self):
                 train_loss = train_loss.mean()
 
                 self.optimizer.zero_grad()
-                pbar.set_description(f"Epoch {epoch},  MSE: {train_loss.item():.6f}")
+                pbar.set_description(f"Epoch {epoch},  Loss: {train_loss.item():.6f}")
                 train_loss.backward()
                 self.optimizer.step()