feat: adding loss vis

pouya-parsa · Jun 8, 2023 · aee00a2 · aee00a2
1 parent cb71a35
commit aee00a2
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -39,6 +39,7 @@ Evaluate the model on validation set:
 accu_val, loss_val = evaluate(model, optimizer, criterion, val_dataloader)
 ``` 
 
-
-
-
+## Visualizing Loss (left: train, right: validation)
+Here is the loss after 30 minutes of training.
+<br/>
+<img src="images/train_loss.png" width="350" /> <img src="images/val_loss.png" width="350" />
diff --git a/images/train_loss.png b/images/train_loss.png
diff --git a/images/val_loss.png b/images/val_loss.png