theluyuan/RWKV-Runner
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add high loss warning

Browse Source
This commit is contained in:
josc146 2023-11-30 12:40:16 +08:00
parent a96d7aef8d
commit 24cc8be085
3 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
frontend/src/_locales/ja/main.json
View File

@ -301,5 +301,6 @@
"Start Time": "開始時間", "Start Time": "開始時間",
"Content Duration": "内容の長さ", "Content Duration": "内容の長さ",
"Please select a MIDI device first": "まずMIDIデバイスを選択してください", "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ドライバが最新であることを確認してください。"
} }

3
frontend/src/_locales/zh-hans/main.json
View File

@ -301,5 +301,6 @@
"Start Time": "开始时间", "Start Time": "开始时间",
"Content Duration": "内容时长", "Content Duration": "内容时长",
"Please select a MIDI device first": "请先选择一个MIDI设备", "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过高，请检查训练数据，并确保你的显卡驱动是最新的"
} }

5
frontend/src/pages/Train.tsx
View File

@ -66,6 +66,11 @@ const parseLossData = (data: string) => {
const loss = parseFloat(lastMatch[8]); const loss = parseFloat(lastMatch[8]);
commonStore.setChartTitle(`Epoch ${epoch}: ${lastMatch[2]} - ${lastMatch[3]}/${lastMatch[4]} - ${lastMatch[5]}/${lastMatch[6]} - ${lastMatch[7]} Loss=${loss}`); commonStore.setChartTitle(`Epoch ${epoch}: ${lastMatch[2]} - ${lastMatch[3]}/${lastMatch[4]} - ${lastMatch[5]}/${lastMatch[6]} - ${lastMatch[7]} Loss=${loss}`);
addLossDataToChart(epoch, 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; return true;
}; };