fixup! discojs-core/models: add gpt

epfml · Mar 15, 2024 · 708d4e7 · 708d4e7
1 parent 1762b10
commit 708d4e7
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/discojs/discojs-core/src/informant/training_informant/base.ts b/discojs/discojs-core/src/informant/training_informant/base.ts
@@ -73,12 +73,15 @@ export abstract class Base {
     return this.validationGraphInformant.accuracy()
   }
 
-  /** add loss for current round */
-  // eslint-disable-next-line accessor-pairs
-  set loss (loss: number) {
+  set loss (loss: number | undefined) {
+    if (loss === undefined) throw new Error('loss is undefined')
     this._losses = this._losses.push(loss)
   }
 
+  get loss (): number | undefined {
+    return this._losses.last()
+  }
+
   /** return loss of each round */
   get losses (): List<number> {
     return this._losses