David19930
commited on
Commit
•
b3202cd
1
Parent(s):
a6611fb
Update README.md
Browse files
README.md
CHANGED
@@ -60,9 +60,9 @@ with torch.no_grad():
|
|
60 |
# Model Performance
|
61 |
- Training Loss: 0.0614
|
62 |
- Validation Loss: 0.1784
|
63 |
-
- Training Dataset: Balloon Dataset (
|
64 |
-
- Validation Dataset: Balloon Dataset (
|
65 |
-
- Number of Epochs:
|
66 |
|
67 |
|
68 |
# Citation
|
|
|
60 |
# Model Performance
|
61 |
- Training Loss: 0.0614
|
62 |
- Validation Loss: 0.1784
|
63 |
+
- Training Dataset: Balloon Dataset (61 images)
|
64 |
+
- Validation Dataset: Balloon Dataset (13 images)
|
65 |
+
- Number of Epochs: 18
|
66 |
|
67 |
|
68 |
# Citation
|