evan/ai-toolkit
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adjust signal amplification target. Allow signal amplification strength in config.

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-03-22 08:30:13 -06:00
1 parent dcd98dc0d5
commit 0f075fc45e
2 files changed
+3
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/config_modules.py
+1
View File
@@ -402,6 +402,7 @@ class TrainConfig:
self.do_batch_noise_correction = kwargs.get('do_batch_noise_correction', False)
self.batch_noise_correction_scale = kwargs.get('batch_noise_correction_scale', 0.1)
self.do_signal_amplification = kwargs.get('do_signal_amplification', False)
self.signal_amplification_strength = kwargs.get('signal_amplification_strength', 0.5)
self.signal_correction_noise_scale = kwargs.get('signal_correction_noise_scale', 1.0)
self.random_noise_shift = kwargs.get('random_noise_shift', 0.0)