Log update epoch 2

train_log.json

@@ -7,12 +7,24 @@
       "prior_loss": 1.0235,
       "diff_loss": 2.3368,
       "time_s": 55.0
+    },
+    {
+      "epoch": 2,
+      "train_loss": 3.6427,
+      "dur_loss": 0.5844,
+      "prior_loss": 0.978,
+      "diff_loss": 2.0803,
+      "time_s": 51.5
     }
   ],
   "val": [
     {
       "epoch": 1,
       "val_loss": 4.0411
+    },
+    {
+      "epoch": 2,
+      "val_loss": 3.9248
     }
   ]
 }