David19930 commited on
Commit
b3202cd
1 Parent(s): a6611fb

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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 (XXX images)
64
- - Validation Dataset: Balloon Dataset (XXX images)
65
- - Number of Epochs: 75
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