stabilize training

2026-08-31 00:10:39 +02:00 · 2025-06-17 13:06:47 +05:30
parent fc766b8327
commit e894e4172a
3 changed files with 19 additions and 7 deletions
@@ -89,9 +89,13 @@ mask_weight = 5.0
 ```
 [training.trainer]
 accumulate_size = 4
-learning_rate = 0.0004
+generator_learning_rate = 0.0004
+discriminator_learning_rate = 0.0002
+momentum = 0.5
 gradient_clip = 20.0
 noise_factor = 0.05
+scheduler_factor = 0.7
+scheduler_patience = 2000
 max_epochs = 50
 strategy = auto
 precision = 16-mixed