Update train_network.py

2026-04-06 21:52:27 +00:00 · 2024-08-04 17:36:34 +08:00
parent aa850aa531
commit cdb2d9c516
1 changed files with 1 additions and 1 deletions
--- a/train_network.py
+++ b/train_network.py
@@ -192,7 +192,7 @@ class NetworkTrainer:
                loss = apply_debiased_estimation(loss, timesteps, noise_scheduler)
                loss = loss.mean()  # 平均なのでbatch_sizeで割る必要なし
                total_loss += loss
-            
+
        average_loss = total_loss / len(timesteps_list)    
        return average_loss