Batch reset

src/store.ts

@@ -166,8 +166,6 @@ function trainingStateReset() {
     setStore('trainingState', {
       forwardStepResults: [],
       backwardStepGradients: [],
-      lossHistory: [],
-      currentLoss: null,
       weightUpdateResults: [],
       currentPhase: 'idle'
     });
@@ -281,6 +279,15 @@ function updateWeights() {
     setStore('networkUpdateTrigger', store.networkUpdateTrigger + 1);
 
     console.log("Action: Weights updated successfully");
+
+    // ** Reset Batch After Weight Update **
+    store.trainer.resetBatch();
+    batch(() => {
+      setStore('trainingState', 'forwardStepResults', []);
+      setStore('trainingState', 'backwardStepGradients', []);
+      setStore('trainingState', 'currentPhase', 'idle');
+    });
+    console.log('Batch has been reset for the next training cycle.');
   });
 }
 

src/trainer.ts

@@ -249,4 +249,11 @@ export class Trainer {
     return this.currentIteration;
   }
 
+  resetBatch(): void {
+    this.currentBatchInputs = [];
+    this.currentBatchTargets = [];
+    this.currentDataIndex = 0;
+    console.log('Trainer batch has been reset.');
+  }
+
 }

src/types.ts

@@ -99,18 +99,12 @@ export interface BackwardStepGradientsPerConnection {
 export type BackwardStepGradients = BackwardStepGradientsPerConnection[];
 
 export interface TrainingStepResult {
-  gradients: number[] | null;
-  oldWeights: number[] | null;
-  newWeights: number[] | null;
+  gradients: BackwardStepGradientsPerConnection[] | null;
+  oldWeights: number[];
+  newWeights: number[];
 }
 
 export interface ForwardStepResults {
   input: number[];
   output: number[];
 }
-
-export interface TrainingStepResult {
-  gradients: number[] | null;
-  oldWeights: number[] | null;
-  newWeights: number[] | null;
-}