Merge basic-flus-branch into update-gh-action

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

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

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

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
     }
 }

results/params.yaml

@@ -1,3 +1,3 @@
 optim_params:
   lr: 0.01
-latent_dim: 8
+latent_dim: 64

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

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

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

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

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

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

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