From 0a62be33a5194b5a56514eb5ef22f02c05b5b9ed Mon Sep 17 00:00:00 2001
From: Kroery <745415461@qq.com>
Date: Wed, 6 Mar 2024 14:10:40 +0800
Subject: [PATCH] update

---
 diffmot.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/diffmot.py b/diffmot.py
index 84cd422..a6f5920 100644
--- 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()