Add stepped loss type

2025-09-22 15:50:12 -06:00
parent 28728a1e92
commit f74475161e
7 changed files with 108 additions and 46 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -123,6 +123,7 @@ export interface TrainConfig {
  diff_output_preservation_multiplier: number;
  diff_output_preservation_class: string;
  switch_boundary_every: number;
+  loss_type: 'mse' | 'mae' | 'wavelet' | 'stepped';
 }

 export interface QuantizeKwargsConfig {