diff --git a/dvc.lock b/dvc.lock
index 2ca2c908a..4eab479b9 100644
--- a/dvc.lock
+++ b/dvc.lock
@@ -31,25 +31,25 @@ stages:
         data_path: fra.txt
         model:
           batch_size: 512
-          latent_dim: 8
+          latent_dim: 64
           duration: 00:00:30:00
-          max_epochs: 6
+          max_epochs: -1
           optim:
             lr: 0.01
-        num_samples: 10000
+        num_samples: 100000
         seed: 423
     outs:
     - path: model
       hash: md5
-      md5: 7006e6dfdcbce27a2362214995586475.dir
-      size: 18561
+      md5: ad9d5090c3792fba67201dfced59d294.dir
+      size: 351617
       nfiles: 1
     - path: results/metrics.json
       hash: md5
-      md5: 098423a9c88c180638b8c06bc5457c53
-      size: 256
+      md5: 563c598564d8e300c4c8728fe064cb59
+      size: 367
     - path: results/plots
       hash: md5
-      md5: ec750cc202d7ea5723c50a8502249e73.dir
-      size: 149
-      nfiles: 5
+      md5: 66cffcad742f7498136fbd5e7149e359.dir
+      size: 2842
+      nfiles: 7
diff --git a/params.yaml b/params.yaml
index af8e523f9..cbd5e365e 100644
--- a/params.yaml
+++ b/params.yaml
@@ -1,10 +1,10 @@
 model:
   batch_size: 512
-  latent_dim: 8
+  latent_dim: 64
   duration: 00:00:30:00
-  max_epochs: 5
+  max_epochs: -1
   optim:
-    lr: 0.001
+    lr: 0.01
 data_path: fra.txt
-num_samples: 10000
+num_samples: 100000
 seed: 423
diff --git a/results/dvc.yaml b/results/dvc.yaml
index 593a913e6..8dc763e4e 100644
--- a/results/dvc.yaml
+++ b/results/dvc.yaml
@@ -7,5 +7,5 @@ plots:
     x: step
 artifacts:
   best:
-    path: ../model/epoch=0-step=16.ckpt
+    path: ../model/epoch=9-step=1505.ckpt
     type: model
diff --git a/results/metrics.json b/results/metrics.json
index 537763f8c..e29bb20dd 100644
--- a/results/metrics.json
+++ b/results/metrics.json
@@ -1,14 +1,18 @@
 {
-    "val": {
-        "loss": 0.871090829372406,
-        "acc": 0.012735245749354362
-    },
-    "epoch": 0,
-    "step": 15,
     "train": {
+        "step": {
+            "loss": 0.14762531220912933,
+            "acc": 0.23080527782440186
+        },
         "epoch": {
-            "loss": 1.7466247081756592,
-            "acc": 0.01202675886452198
+            "loss": 0.14830222725868225,
+            "acc": 0.2522057294845581
         }
+    },
+    "epoch": 9,
+    "step": 1504,
+    "val": {
+        "loss": 0.14832937717437744,
+        "acc": 0.2539016008377075
     }
 }
diff --git a/results/params.yaml b/results/params.yaml
index 7a4fb2e19..d7bbc6397 100644
--- a/results/params.yaml
+++ b/results/params.yaml
@@ -1,3 +1,3 @@
 optim_params:
   lr: 0.01
-latent_dim: 8
+latent_dim: 64
diff --git a/results/plots/metrics/epoch.tsv b/results/plots/metrics/epoch.tsv
index 9c404b6cb..e5b5ac8a3 100644
--- a/results/plots/metrics/epoch.tsv
+++ b/results/plots/metrics/epoch.tsv
@@ -1,3 +1,52 @@
 step	epoch
-15	0
-15	0
+49	0
+99	0
+149	0
+156	0
+156	0
+199	1
+249	1
+299	1
+313	1
+313	1
+349	2
+399	2
+449	2
+470	2
+470	2
+499	3
+549	3
+599	3
+627	3
+627	3
+649	4
+699	4
+749	4
+784	4
+784	4
+799	5
+849	5
+899	5
+941	5
+941	5
+949	6
+999	6
+1049	6
+1098	6
+1098	6
+1099	7
+1149	7
+1199	7
+1249	7
+1255	7
+1255	7
+1299	8
+1349	8
+1399	8
+1412	8
+1412	8
+1449	9
+1499	9
+1504	9
+1504	9
+1504	9
diff --git a/results/plots/metrics/train/epoch/acc.tsv b/results/plots/metrics/train/epoch/acc.tsv
index 366d6504d..b168b10be 100644
--- a/results/plots/metrics/train/epoch/acc.tsv
+++ b/results/plots/metrics/train/epoch/acc.tsv
@@ -1,2 +1,11 @@
 step	acc
-15	0.01202675886452198
+156	0.09887676686048508
+313	0.1471019834280014
+470	0.18248222768306732
+627	0.20411770045757294
+784	0.2190815657377243
+941	0.2287890911102295
+1098	0.23663000762462616
+1255	0.24443037807941437
+1412	0.2499464601278305
+1504	0.2522057294845581
diff --git a/results/plots/metrics/train/epoch/loss.tsv b/results/plots/metrics/train/epoch/loss.tsv
index b78bef5f1..dd9a19803 100644
--- a/results/plots/metrics/train/epoch/loss.tsv
+++ b/results/plots/metrics/train/epoch/loss.tsv
@@ -1,2 +1,11 @@
 step	loss
-15	1.7466247081756592
+156	0.2664768397808075
+313	0.1885426789522171
+470	0.1741955429315567
+627	0.1659986525774002
+784	0.16037209331989288
+941	0.1562994122505188
+1098	0.15349090099334717
+1255	0.15124300122261047
+1412	0.14945325255393982
+1504	0.14830222725868225
diff --git a/results/plots/metrics/train/step/acc.tsv b/results/plots/metrics/train/step/acc.tsv
new file mode 100644
index 000000000..1dc384a6d
--- /dev/null
+++ b/results/plots/metrics/train/step/acc.tsv
@@ -0,0 +1,32 @@
+step	acc
+49	0.09062056243419647
+99	0.10510364919900894
+149	0.13709327578544617
+199	0.1394631266593933
+249	0.14277693629264832
+299	0.16892829537391663
+349	0.17652598023414612
+399	0.1577042043209076
+449	0.1870027780532837
+499	0.19452029466629028
+549	0.21528783440589905
+599	0.23040702939033508
+649	0.19968083500862122
+699	0.21933883428573608
+749	0.20493647456169128
+799	0.22558006644248962
+849	0.23612238466739655
+899	0.21324488520622253
+949	0.2304501235485077
+999	0.22250041365623474
+1049	0.23441341519355774
+1099	0.23782780766487122
+1149	0.24050183594226837
+1199	0.2548307776451111
+1249	0.22360961139202118
+1299	0.24851763248443604
+1349	0.2732715904712677
+1399	0.2399168312549591
+1449	0.22618699073791504
+1499	0.2636773884296417
+1504	0.23080527782440186
diff --git a/results/plots/metrics/train/step/loss.tsv b/results/plots/metrics/train/step/loss.tsv
new file mode 100644
index 000000000..04a25c033
--- /dev/null
+++ b/results/plots/metrics/train/step/loss.tsv
@@ -0,0 +1,32 @@
+step	loss
+49	0.23360593616962433
+99	0.21666863560676575
+149	0.20438745617866516
+199	0.18772511184215546
+249	0.17944981157779694
+299	0.18668335676193237
+349	0.17598168551921844
+399	0.16521920263767242
+449	0.16728925704956055
+499	0.16383859515190125
+549	0.1687503457069397
+599	0.16121630370616913
+649	0.1566413938999176
+699	0.16148604452610016
+749	0.16010449826717377
+799	0.15542428195476532
+849	0.15153297781944275
+899	0.14997026324272156
+949	0.15178194642066956
+999	0.15090182423591614
+1049	0.1520526111125946
+1099	0.16291281580924988
+1149	0.14956119656562805
+1199	0.1448971927165985
+1249	0.1551576554775238
+1299	0.1513485610485077
+1349	0.1515946090221405
+1399	0.14243841171264648
+1449	0.1564590185880661
+1499	0.14867596328258514
+1504	0.14762531220912933
diff --git a/results/plots/metrics/val/acc.tsv b/results/plots/metrics/val/acc.tsv
new file mode 100644
index 000000000..14d0b6db8
--- /dev/null
+++ b/results/plots/metrics/val/acc.tsv
@@ -0,0 +1,11 @@
+step	acc
+156	0.13435418903827667
+313	0.1657787412405014
+470	0.19466444849967957
+627	0.21133434772491455
+784	0.22303569316864014
+941	0.23581743240356445
+1098	0.24472711980342865
+1255	0.24441975355148315
+1412	0.2471277415752411
+1504	0.2539016008377075
diff --git a/results/plots/metrics/val/loss.tsv b/results/plots/metrics/val/loss.tsv
new file mode 100644
index 000000000..a23d8fa54
--- /dev/null
+++ b/results/plots/metrics/val/loss.tsv
@@ -0,0 +1,11 @@
+step	loss
+156	0.2001143991947174
+313	0.18106751143932343
+470	0.17055320739746094
+627	0.16768312454223633
+784	0.16413633525371552
+941	0.16012321412563324
+1098	0.1558203548192978
+1255	0.15428794920444489
+1412	0.15231744945049286
+1504	0.14832937717437744
diff --git a/results/report.html b/results/report.html
index 56e441a14..5ea188892 100644
--- a/results/report.html
+++ b/results/report.html
@@ -25,7 +25,7 @@
 <tr><th style="text-align: right;">  optim_params.lr</th><th style="text-align: right;">  latent_dim</th></tr>
 </thead>
 <tbody>
-<tr><td style="text-align: right;">             0.01</td><td style="text-align: right;">           8</td></tr>
+<tr><td style="text-align: right;">             0.01</td><td style="text-align: right;">          64</td></tr>
 </tbody>
 </table>
             </div>
@@ -36,10 +36,10 @@
             <div style="display: flex;justify-content: center;">
                 <table>
 <thead>
-<tr><th style="text-align: right;">  val.loss</th><th style="text-align: right;">  val.acc</th><th style="text-align: right;">  epoch</th><th style="text-align: right;">  step</th><th style="text-align: right;">  train.epoch.loss</th><th style="text-align: right;">  train.epoch.acc</th></tr>
+<tr><th style="text-align: right;">  train.step.loss</th><th style="text-align: right;">  train.step.acc</th><th style="text-align: right;">  train.epoch.loss</th><th style="text-align: right;">  train.epoch.acc</th><th style="text-align: right;">  epoch</th><th style="text-align: right;">  step</th><th style="text-align: right;">  val.loss</th><th style="text-align: right;">  val.acc</th></tr>
 </thead>
 <tbody>
-<tr><td style="text-align: right;">  0.871091</td><td style="text-align: right;">0.0127352</td><td style="text-align: right;">      0</td><td style="text-align: right;">    15</td><td style="text-align: right;">           1.74662</td><td style="text-align: right;">        0.0120268</td></tr>
+<tr><td style="text-align: right;">         0.147625</td><td style="text-align: right;">        0.230805</td><td style="text-align: right;">          0.148302</td><td style="text-align: right;">         0.252206</td><td style="text-align: right;">      9</td><td style="text-align: right;">  1504</td><td style="text-align: right;">  0.148329</td><td style="text-align: right;"> 0.253902</td></tr>
 </tbody>
 </table>
             </div>
@@ -47,40 +47,56 @@
 
     <div id = "static_epoch">
         <script type = "text/javascript">
-            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "15", "epoch": "0", "rev": "workspace"}, {"step": "15", "epoch": "0", "rev": "workspace"}]}, "title": "epoch", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "epoch", "title": "epoch", "type": "quantitative"}]}, "transform": [{"loess": "epoch", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "epoch", "title": "epoch", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "49", "epoch": "0", "rev": "workspace"}, {"step": "99", "epoch": "0", "rev": "workspace"}, {"step": "149", "epoch": "0", "rev": "workspace"}, {"step": "156", "epoch": "0", "rev": "workspace"}, {"step": "156", "epoch": "0", "rev": "workspace"}, {"step": "199", "epoch": "1", "rev": "workspace"}, {"step": "249", "epoch": "1", "rev": "workspace"}, {"step": "299", "epoch": "1", "rev": "workspace"}, {"step": "313", "epoch": "1", "rev": "workspace"}, {"step": "313", "epoch": "1", "rev": "workspace"}, {"step": "349", "epoch": "2", "rev": "workspace"}, {"step": "399", "epoch": "2", "rev": "workspace"}, {"step": "449", "epoch": "2", "rev": "workspace"}, {"step": "470", "epoch": "2", "rev": "workspace"}, {"step": "470", "epoch": "2", "rev": "workspace"}, {"step": "499", "epoch": "3", "rev": "workspace"}, {"step": "549", "epoch": "3", "rev": "workspace"}, {"step": "599", "epoch": "3", "rev": "workspace"}, {"step": "627", "epoch": "3", "rev": "workspace"}, {"step": "627", "epoch": "3", "rev": "workspace"}, {"step": "649", "epoch": "4", "rev": "workspace"}, {"step": "699", "epoch": "4", "rev": "workspace"}, {"step": "749", "epoch": "4", "rev": "workspace"}, {"step": "784", "epoch": "4", "rev": "workspace"}, {"step": "784", "epoch": "4", "rev": "workspace"}, {"step": "799", "epoch": "5", "rev": "workspace"}, {"step": "849", "epoch": "5", "rev": "workspace"}, {"step": "899", "epoch": "5", "rev": "workspace"}, {"step": "941", "epoch": "5", "rev": "workspace"}, {"step": "941", "epoch": "5", "rev": "workspace"}, {"step": "949", "epoch": "6", "rev": "workspace"}, {"step": "999", "epoch": "6", "rev": "workspace"}, {"step": "1049", "epoch": "6", "rev": "workspace"}, {"step": "1098", "epoch": "6", "rev": "workspace"}, {"step": "1098", "epoch": "6", "rev": "workspace"}, {"step": "1099", "epoch": "7", "rev": "workspace"}, {"step": "1149", "epoch": "7", "rev": "workspace"}, {"step": "1199", "epoch": "7", "rev": "workspace"}, {"step": "1249", "epoch": "7", "rev": "workspace"}, {"step": "1255", "epoch": "7", "rev": "workspace"}, {"step": "1255", "epoch": "7", "rev": "workspace"}, {"step": "1299", "epoch": "8", "rev": "workspace"}, {"step": "1349", "epoch": "8", "rev": "workspace"}, {"step": "1399", "epoch": "8", "rev": "workspace"}, {"step": "1412", "epoch": "8", "rev": "workspace"}, {"step": "1412", "epoch": "8", "rev": "workspace"}, {"step": "1449", "epoch": "9", "rev": "workspace"}, {"step": "1499", "epoch": "9", "rev": "workspace"}, {"step": "1504", "epoch": "9", "rev": "workspace"}, {"step": "1504", "epoch": "9", "rev": "workspace"}, {"step": "1504", "epoch": "9", "rev": "workspace"}]}, "title": "epoch", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "epoch", "title": "epoch", "type": "quantitative"}]}, "transform": [{"loess": "epoch", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "epoch", "title": "epoch", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "epoch", "type": "quantitative", "title": "epoch", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
             vegaEmbed('#static_epoch', spec);
         </script>
     </div>
     
 
-    <div id = "static_train_epoch_loss">
+    <div id = "static_val_loss">
         <script type = "text/javascript">
-            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "15", "loss": "1.7466247081756592", "rev": "workspace"}]}, "title": "train/epoch/loss", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}, "transform": [{"loess": "loss", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
-            vegaEmbed('#static_train_epoch_loss', spec);
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "156", "loss": "0.2001143991947174", "rev": "workspace"}, {"step": "313", "loss": "0.18106751143932343", "rev": "workspace"}, {"step": "470", "loss": "0.17055320739746094", "rev": "workspace"}, {"step": "627", "loss": "0.16768312454223633", "rev": "workspace"}, {"step": "784", "loss": "0.16413633525371552", "rev": "workspace"}, {"step": "941", "loss": "0.16012321412563324", "rev": "workspace"}, {"step": "1098", "loss": "0.1558203548192978", "rev": "workspace"}, {"step": "1255", "loss": "0.15428794920444489", "rev": "workspace"}, {"step": "1412", "loss": "0.15231744945049286", "rev": "workspace"}, {"step": "1504", "loss": "0.14832937717437744", "rev": "workspace"}]}, "title": "val/loss", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}, "transform": [{"loess": "loss", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_val_loss', spec);
         </script>
     </div>
     
 
-    <div id = "static_train_epoch_acc">
+    <div id = "static_val_acc">
         <script type = "text/javascript">
-            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "15", "acc": "0.01202675886452198", "rev": "workspace"}]}, "title": "train/epoch/acc", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}, "transform": [{"loess": "acc", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
-            vegaEmbed('#static_train_epoch_acc', spec);
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "156", "acc": "0.13435418903827667", "rev": "workspace"}, {"step": "313", "acc": "0.1657787412405014", "rev": "workspace"}, {"step": "470", "acc": "0.19466444849967957", "rev": "workspace"}, {"step": "627", "acc": "0.21133434772491455", "rev": "workspace"}, {"step": "784", "acc": "0.22303569316864014", "rev": "workspace"}, {"step": "941", "acc": "0.23581743240356445", "rev": "workspace"}, {"step": "1098", "acc": "0.24472711980342865", "rev": "workspace"}, {"step": "1255", "acc": "0.24441975355148315", "rev": "workspace"}, {"step": "1412", "acc": "0.2471277415752411", "rev": "workspace"}, {"step": "1504", "acc": "0.2539016008377075", "rev": "workspace"}]}, "title": "val/acc", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}, "transform": [{"loess": "acc", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_val_acc', spec);
         </script>
     </div>
     
 
-    <div id = "static_val_loss">
+    <div id = "static_train_step_loss">
         <script type = "text/javascript">
-            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "15", "loss": "0.871090829372406", "rev": "workspace"}]}, "title": "val/loss", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}, "transform": [{"loess": "loss", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
-            vegaEmbed('#static_val_loss', spec);
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "49", "loss": "0.23360593616962433", "rev": "workspace"}, {"step": "99", "loss": "0.21666863560676575", "rev": "workspace"}, {"step": "149", "loss": "0.20438745617866516", "rev": "workspace"}, {"step": "199", "loss": "0.18772511184215546", "rev": "workspace"}, {"step": "249", "loss": "0.17944981157779694", "rev": "workspace"}, {"step": "299", "loss": "0.18668335676193237", "rev": "workspace"}, {"step": "349", "loss": "0.17598168551921844", "rev": "workspace"}, {"step": "399", "loss": "0.16521920263767242", "rev": "workspace"}, {"step": "449", "loss": "0.16728925704956055", "rev": "workspace"}, {"step": "499", "loss": "0.16383859515190125", "rev": "workspace"}, {"step": "549", "loss": "0.1687503457069397", "rev": "workspace"}, {"step": "599", "loss": "0.16121630370616913", "rev": "workspace"}, {"step": "649", "loss": "0.1566413938999176", "rev": "workspace"}, {"step": "699", "loss": "0.16148604452610016", "rev": "workspace"}, {"step": "749", "loss": "0.16010449826717377", "rev": "workspace"}, {"step": "799", "loss": "0.15542428195476532", "rev": "workspace"}, {"step": "849", "loss": "0.15153297781944275", "rev": "workspace"}, {"step": "899", "loss": "0.14997026324272156", "rev": "workspace"}, {"step": "949", "loss": "0.15178194642066956", "rev": "workspace"}, {"step": "999", "loss": "0.15090182423591614", "rev": "workspace"}, {"step": "1049", "loss": "0.1520526111125946", "rev": "workspace"}, {"step": "1099", "loss": "0.16291281580924988", "rev": "workspace"}, {"step": "1149", "loss": "0.14956119656562805", "rev": "workspace"}, {"step": "1199", "loss": "0.1448971927165985", "rev": "workspace"}, {"step": "1249", "loss": "0.1551576554775238", "rev": "workspace"}, {"step": "1299", "loss": "0.1513485610485077", "rev": "workspace"}, {"step": "1349", "loss": "0.1515946090221405", "rev": "workspace"}, {"step": "1399", "loss": "0.14243841171264648", "rev": "workspace"}, {"step": "1449", "loss": "0.1564590185880661", "rev": "workspace"}, {"step": "1499", "loss": "0.14867596328258514", "rev": "workspace"}, {"step": "1504", "loss": "0.14762531220912933", "rev": "workspace"}]}, "title": "train/step/loss", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}, "transform": [{"loess": "loss", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_train_step_loss', spec);
         </script>
     </div>
     
 
-    <div id = "static_val_acc">
+    <div id = "static_train_step_acc">
         <script type = "text/javascript">
-            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "15", "acc": "0.012735245749354362", "rev": "workspace"}]}, "title": "val/acc", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}, "transform": [{"loess": "acc", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
-            vegaEmbed('#static_val_acc', spec);
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "49", "acc": "0.09062056243419647", "rev": "workspace"}, {"step": "99", "acc": "0.10510364919900894", "rev": "workspace"}, {"step": "149", "acc": "0.13709327578544617", "rev": "workspace"}, {"step": "199", "acc": "0.1394631266593933", "rev": "workspace"}, {"step": "249", "acc": "0.14277693629264832", "rev": "workspace"}, {"step": "299", "acc": "0.16892829537391663", "rev": "workspace"}, {"step": "349", "acc": "0.17652598023414612", "rev": "workspace"}, {"step": "399", "acc": "0.1577042043209076", "rev": "workspace"}, {"step": "449", "acc": "0.1870027780532837", "rev": "workspace"}, {"step": "499", "acc": "0.19452029466629028", "rev": "workspace"}, {"step": "549", "acc": "0.21528783440589905", "rev": "workspace"}, {"step": "599", "acc": "0.23040702939033508", "rev": "workspace"}, {"step": "649", "acc": "0.19968083500862122", "rev": "workspace"}, {"step": "699", "acc": "0.21933883428573608", "rev": "workspace"}, {"step": "749", "acc": "0.20493647456169128", "rev": "workspace"}, {"step": "799", "acc": "0.22558006644248962", "rev": "workspace"}, {"step": "849", "acc": "0.23612238466739655", "rev": "workspace"}, {"step": "899", "acc": "0.21324488520622253", "rev": "workspace"}, {"step": "949", "acc": "0.2304501235485077", "rev": "workspace"}, {"step": "999", "acc": "0.22250041365623474", "rev": "workspace"}, {"step": "1049", "acc": "0.23441341519355774", "rev": "workspace"}, {"step": "1099", "acc": "0.23782780766487122", "rev": "workspace"}, {"step": "1149", "acc": "0.24050183594226837", "rev": "workspace"}, {"step": "1199", "acc": "0.2548307776451111", "rev": "workspace"}, {"step": "1249", "acc": "0.22360961139202118", "rev": "workspace"}, {"step": "1299", "acc": "0.24851763248443604", "rev": "workspace"}, {"step": "1349", "acc": "0.2732715904712677", "rev": "workspace"}, {"step": "1399", "acc": "0.2399168312549591", "rev": "workspace"}, {"step": "1449", "acc": "0.22618699073791504", "rev": "workspace"}, {"step": "1499", "acc": "0.2636773884296417", "rev": "workspace"}, {"step": "1504", "acc": "0.23080527782440186", "rev": "workspace"}]}, "title": "train/step/acc", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}, "transform": [{"loess": "acc", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_train_step_acc', spec);
+        </script>
+    </div>
+    
+
+    <div id = "static_train_epoch_loss">
+        <script type = "text/javascript">
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "156", "loss": "0.2664768397808075", "rev": "workspace"}, {"step": "313", "loss": "0.1885426789522171", "rev": "workspace"}, {"step": "470", "loss": "0.1741955429315567", "rev": "workspace"}, {"step": "627", "loss": "0.1659986525774002", "rev": "workspace"}, {"step": "784", "loss": "0.16037209331989288", "rev": "workspace"}, {"step": "941", "loss": "0.1562994122505188", "rev": "workspace"}, {"step": "1098", "loss": "0.15349090099334717", "rev": "workspace"}, {"step": "1255", "loss": "0.15124300122261047", "rev": "workspace"}, {"step": "1412", "loss": "0.14945325255393982", "rev": "workspace"}, {"step": "1504", "loss": "0.14830222725868225", "rev": "workspace"}]}, "title": "train/epoch/loss", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}, "transform": [{"loess": "loss", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "loss", "title": "loss", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "loss", "type": "quantitative", "title": "loss", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_train_epoch_loss', spec);
+        </script>
+    </div>
+    
+
+    <div id = "static_train_epoch_acc">
+        <script type = "text/javascript">
+            var spec = {"$schema": "https://vega.github.io/schema/vega-lite/v5.json", "data": {"values": [{"step": "156", "acc": "0.09887676686048508", "rev": "workspace"}, {"step": "313", "acc": "0.1471019834280014", "rev": "workspace"}, {"step": "470", "acc": "0.18248222768306732", "rev": "workspace"}, {"step": "627", "acc": "0.20411770045757294", "rev": "workspace"}, {"step": "784", "acc": "0.2190815657377243", "rev": "workspace"}, {"step": "941", "acc": "0.2287890911102295", "rev": "workspace"}, {"step": "1098", "acc": "0.23663000762462616", "rev": "workspace"}, {"step": "1255", "acc": "0.24443037807941437", "rev": "workspace"}, {"step": "1412", "acc": "0.2499464601278305", "rev": "workspace"}, {"step": "1504", "acc": "0.2522057294845581", "rev": "workspace"}]}, "title": "train/epoch/acc", "width": 300, "height": 300, "params": [{"name": "smooth", "value": 0.001, "bind": {"input": "range", "min": 0.001, "max": 1, "step": 0.001}}], "layer": [{"mark": "line", "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}, "transform": [{"loess": "acc", "on": "step", "groupby": ["rev", "filename", "field", "filename::field"], "bandwidth": {"signal": "smooth"}}]}, {"mark": {"type": "line", "opacity": 0.2}, "encoding": {"x": {"field": "step", "type": "quantitative", "title": "step"}, "y": {"field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}, "tooltip": [{"field": "step", "title": "step", "type": "quantitative"}, {"field": "acc", "title": "acc", "type": "quantitative"}]}}, {"mark": {"type": "circle", "size": 10, "tooltip": {"content": "encoding"}}, "encoding": {"x": {"aggregate": "max", "field": "step", "type": "quantitative", "title": "step"}, "y": {"aggregate": {"argmax": "step"}, "field": "acc", "type": "quantitative", "title": "acc", "scale": {"zero": false}}, "color": {"field": "rev", "type": "nominal"}}}]};
+            vegaEmbed('#static_train_epoch_acc', spec);
         </script>
     </div>