Smolry commited on
Commit
e29b18b
·
verified ·
1 Parent(s): 12184ee

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -6
README.md CHANGED
@@ -219,16 +219,21 @@ Test Samples:
219
 
220
  ## Training History
221
 
222
- Training curves are available in:
223
 
224
- training/
 
 
 
 
 
225
 
226
- Contents:
227
 
228
- - train_loss.png
229
- - validation_loss.png
230
- - learning_curve.png
231
 
 
 
232
  ---
233
 
234
  ## Explainability
 
219
 
220
  ## Training History
221
 
222
+ The model was trained in two stages:
223
 
224
+ 1. Classifier training with the EfficientNet-B0 backbone frozen.
225
+ 2. Fine-tuning of the final feature blocks.
226
+
227
+ ### Phase 1
228
+
229
+ ![Phase 1 Training Loss](training/phase1_loss.png)
230
 
231
+ ### Phase 2
232
 
233
+ ![Phase 2 Fine-Tuning Loss](training/phase2_loss.png)
 
 
234
 
235
+ The epoch-level training history is available in
236
+ `training/training_history.json`.
237
  ---
238
 
239
  ## Explainability