10 epoch training with evaluation

dvc.lock

@@ -23,8 +23,8 @@ train:
     md5: c94ea029ed76ca94bb1ad4c1655e5e68
     size: 1916
   - path: src/code/params.yml
-    md5: 2263ca2167c1bb4b0f53a9aedb5f238e
-    size: 217
+    md5: f1b2e4ff61e3f3fd8315f6c2c093bd49
+    size: 218
   - path: src/code/training.py
     md5: e3dff7f4b59e4ebf818d7631d3e6803a
     size: 1683
@@ -34,14 +34,14 @@ train:
     nfiles: 1590
   outs:
   - path: logs/train_metrics.csv
-    md5: 437a06e6c6c5b4f6eec5e546c1ce6930
-    size: 103916
+    md5: f9c97a7224444def38e5df5ecfd3d55e
+    size: 1256463
   - path: logs/train_params.yml
-    md5: e06e92ac0f3ac1d367c22a10c28cccf9
-    size: 886
+    md5: d3a44d36bffe5e02f2c0694446bd06fc
+    size: 887
   - path: src/models/
-    md5: fab42526c433987e0e6370db31a1869d.dir
-    size: 494927196
+    md5: a6a861609f11cfd8365a49a27482729a.dir
+    size: 494927324
     nfiles: 1
 eval:
   cmd: python3 src/code/eval.py src/data/processed/test
@@ -56,20 +56,20 @@ eval:
     md5: 2fc866e1107042a996087d5716d44bf0
     size: 2999
   - path: src/code/params.yml
-    md5: 2263ca2167c1bb4b0f53a9aedb5f238e
-    size: 217
+    md5: f1b2e4ff61e3f3fd8315f6c2c093bd49
+    size: 218
   - path: src/data/processed/test
     md5: bcccd66f3f561b53ba97c89a558c08a0.dir
     size: 88596370
     nfiles: 1308
   - path: src/models/model.pth
-    md5: 2fd77305fd779eefd11e307ee3f201d7
-    size: 494927196
+    md5: e296d850af2e0a6c49173e239f75896f
+    size: 494927324
   outs:
   - path: logs/test_metrics.csv
-    md5: 80d51ef1a70b1314947919a5c37f3220
-    size: 341
+    md5: b0518e1d7d407223b34895549353ad7e
+    size: 342
   - path: src/eval/examples/
-    md5: cc137b33b5f5930e304b0b65f4b546ca.dir
-    size: 759235
+    md5: 57b73ac5d551f2706d4250842028ba85.dir
+    size: 644233
     nfiles: 10

logs/test_metrics.csv

@@ -1,10 +1,10 @@
 Name,Value,Timestamp,Step
-"a1",0.5386207,1615290337662,1
-"a2",0.8303751,1615290337662,1
-"a3",0.93854135,1615290337662,1
-"abs_rel",0.29692116,1615290337662,1
-"sq_rel",0.32333422,1615290337662,1
-"rmse",0.848144,1615290337662,1
-"rmse_log",0.32479692,1615290337662,1
-"log10",0.11296987,1615290337662,1
-"silog",27.511873,1615290337662,1
+"a1",0.6427222,1615291294320,1
+"a2",0.8990741,1615291294320,1
+"a3",0.97092867,1615291294320,1
+"abs_rel",0.22883615,1615291294320,1
+"sq_rel",0.21042906,1615291294320,1
+"rmse",0.6880537,1615291294320,1
+"rmse_log",0.2589324,1615291294320,1
+"log10",0.089840144,1615291294320,1
+"silog",21.828184,1615291294320,1

logs/train_params.yml

@@ -1,5 +1,5 @@
 DAGsHubLogger: true
-Learner: <fastai.learner.Learner object at 0x7f051ecfcac8>
+Learner: <fastai.learner.Learner object at 0x7fcf9aec1828>
 ParamScheduler: true
 ProgressCallback: true
 Recorder: {add_time: true, train_metrics: false, valid_metrics: true}
@@ -14,8 +14,8 @@ dls.after_batch: "Pipeline: IntToFloatTensor -- {'div': 255.0, 'div_mask': 1} ->
   \     [[0.2250]]]], device='cuda:0'), 'axes': (0, 2, 3)}"
 dls.after_item: 'Pipeline: ToTensor'
 dls.before_batch: 'Pipeline: '
-frozen: true
-frozen idx: 2
+frozen: false
+frozen idx: 0
 input 1 dim 1: 4
 input 1 dim 2: 3
 input 1 dim 3: 480

src/code/params.yml

@@ -4,7 +4,7 @@ batch_size: 4
 num_workers: 0
 weight_decay: 1e-2
 learning_rate: 1e-3
-epochs: 1
+epochs: 10
 num_outs: 3
 source_dir: src
 model_dir: models