fix loss parser

josStorer · Jul 7, 2023 · c725d11 · c725d11
1 parent ba2a6bd
commit c725d11
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/frontend/src/pages/Train.tsx b/frontend/src/pages/Train.tsx
@@ -49,7 +49,7 @@ ChartJS.register(
 );
 
 const parseLossData = (data: string) => {
- const regex = /Epoch (\d+):\s+(\d+%)\|[\s\S]*\| (\d+)\/(\d+) \[(\d+:\d+)<(\d+:\d+),\s+(\d+.\d+it\/s), loss=(\d+.\d+),[\s\S]*\]/g;
+ const regex = /Epoch (\d+):\s+(\d+%)\|[\s\S]*\| (\d+)\/(\d+) \[(\d+:\d+)<(\d+:\d+),\s+(\d+.\d+it\/s), loss=(\S+),[\s\S]*\]/g;
  const matches = Array.from(data.matchAll(regex));
  if (matches.length === 0)
  return;