tlphams commited on
Commit
5e7a651
1 Parent(s): 44937fd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -73,7 +73,7 @@ The following hyperparameters were used during training:
73
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
74
  - lr_scheduler_type: linear
75
  - num_epochs: 8
76
- - stop_at_epoch: 4
77
 
78
  ### Training results
79
 
 
73
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
74
  - lr_scheduler_type: linear
75
  - num_epochs: 8
76
+ - saved_checkpoint_at_epoch: 4 (condition: loss < 0.3)
77
 
78
  ### Training results
79