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

Added experimental band loss

Browse Source
This commit is contained in:
Jaret Burkett committed 2026-10-15 08:26:12 -06:00
1 parent 7045370346
commit 831ccfa7aa
4 files changed
+160 -2
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

toolkit/config_modules.py
+1 -1
View File
@@ -522,7 +522,7 @@ class TrainConfig:
self.correct_pred_norm = kwargs.get('correct_pred_norm', False)
self.correct_pred_norm_multiplier = kwargs.get('correct_pred_norm_multiplier', 1.0)
self.loss_type = kwargs.get('loss_type', 'mse') # mse, mae, wavelet, pixelspace, mean_flow, pseudo_huber
self.loss_type = kwargs.get('loss_type', 'mse') # mse, mae, wavelet, band, pixelspace, mean_flow, pseudo_huber
# do the loss on a timestep to 0 prediction
self.t0_loss_target = kwargs.get('t0_loss_target', False)