From 24cc8be085e9601e90de48f196e898d48fd98739 Mon Sep 17 00:00:00 2001
From: josc146 <josStorer@outlook.com>
Date: Thu, 30 Nov 2023 12:40:16 +0800
Subject: [PATCH] add high loss warning

---
 frontend/src/_locales/ja/main.json      | 3 ++-
 frontend/src/_locales/zh-hans/main.json | 3 ++-
 frontend/src/pages/Train.tsx            | 5 +++++
 3 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/frontend/src/_locales/ja/main.json b/frontend/src/_locales/ja/main.json
index c89466c..b6befe2 100644
--- a/frontend/src/_locales/ja/main.json
+++ b/frontend/src/_locales/ja/main.json
@@ -301,5 +301,6 @@
   "Start Time": "開始時間",
   "Content Duration": "内容の長さ",
   "Please select a MIDI device first": "まずMIDIデバイスを選択してください",
-  "Piano is the main instrument": "ピアノはメインの楽器です"
+  "Piano is the main instrument": "ピアノはメインの楽器です",
+  "Loss is too high, please check the training data, and ensure your gpu driver is up to date.": "Lossが大きすぎます、トレーニングデータを確認し、GPUドライバが最新であることを確認してください。"
 }
\ No newline at end of file
diff --git a/frontend/src/_locales/zh-hans/main.json b/frontend/src/_locales/zh-hans/main.json
index 9b9659a..105961e 100644
--- a/frontend/src/_locales/zh-hans/main.json
+++ b/frontend/src/_locales/zh-hans/main.json
@@ -301,5 +301,6 @@
   "Start Time": "开始时间",
   "Content Duration": "内容时长",
   "Please select a MIDI device first": "请先选择一个MIDI设备",
-  "Piano is the main instrument": "钢琴为主"
+  "Piano is the main instrument": "钢琴为主",
+  "Loss is too high, please check the training data, and ensure your gpu driver is up to date.": "Loss过高，请检查训练数据，并确保你的显卡驱动是最新的"
 }
\ No newline at end of file
diff --git a/frontend/src/pages/Train.tsx b/frontend/src/pages/Train.tsx
index 31d2306..ba8e9ea 100644
--- a/frontend/src/pages/Train.tsx
+++ b/frontend/src/pages/Train.tsx
@@ -66,6 +66,11 @@ const parseLossData = (data: string) => {
   const loss = parseFloat(lastMatch[8]);
   commonStore.setChartTitle(`Epoch ${epoch}: ${lastMatch[2]} - ${lastMatch[3]}/${lastMatch[4]} - ${lastMatch[5]}/${lastMatch[6]} - ${lastMatch[7]} Loss=${loss}`);
   addLossDataToChart(epoch, loss);
+  if (loss > 5)
+    toast(t('Loss is too high, please check the training data, and ensure your gpu driver is up to date.'), {
+      type: 'warning',
+      toastId: 'train_loss_high'
+    });
   return true;
 };