Added Differential Output Preservation Loss to trainer and ui

2025-02-25 20:12:36 -07:00
parent 259ded9602
commit f6e16e582a
6 changed files with 127 additions and 6 deletions
--- a/ui/src/types.ts
+++ b/ui/src/types.ts
@@ -100,6 +100,9 @@ export interface TrainConfig {
  optimizer_params: {
    weight_decay: number;
  };
+  diff_output_preservation: boolean;
+  diff_output_preservation_multiplier: number;
+  diff_output_preservation_class: string;
 }

 export interface QuantizeKwargsConfig {