Added new experimental time step weighing that should solve a lot of issues with distribution. Updated example. Removed a warning

2024-08-13 12:02:11 -06:00
parent 9ee1ef2a0a
commit 418f5f7e8c
3 changed files with 19 additions and 20 deletions
--- a/toolkit/train_tools.py
+++ b/toolkit/train_tools.py
@@ -640,8 +640,8 @@ def add_all_snr_to_noise_scheduler(noise_scheduler, device):
            all_snr.requires_grad = False
        noise_scheduler.all_snr = all_snr.to(device)
    except Exception as e:
-        print(e)
-        print("Failed to add all_snr to noise_scheduler")
+        # just move on
+        pass


 def get_all_snr(noise_scheduler, device):