webapp/trainer: avoid graph shakes

epfml · Jul 2, 2024 · a0f54a1 · a0f54a1
1 parent 06efe2c
commit a0f54a1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/webapp/src/components/training/Trainer.vue b/webapp/src/components/training/Trainer.vue
@@ -155,17 +155,19 @@ async function startTraining(distributed: boolean): Promise<void> {
       const [roundGen, roundLogs] = async_iterator.split(round)
 
       roundGenerator.value = roundGen
-      epochsOfRoundLogs.value = List<EpochLogs>()
       for await (const epoch of roundGenerator.value) {
         const [epochGen, epochLogs] = async_iterator.split(epoch)
 
         epochGenerator.value = epochGen
-        batchesOfEpochLogs.value = List<BatchLogs>()
         for await (const batch of epochGenerator.value)
           batchesOfEpochLogs.value = batchesOfEpochLogs.value.push(batch);
+
         epochsOfRoundLogs.value = epochsOfRoundLogs.value.push(await epochLogs)
+        batchesOfEpochLogs.value = List<BatchLogs>()
       }
+
       roundsLogs.value = roundsLogs.value.push(await roundLogs)
+      epochsOfRoundLogs.value = List<EpochLogs>()
     }
   } catch (e) {
     if (e === stopper) {