tlphams commited on
Commit
bea9fec
1 Parent(s): 3eb7569

Update README.md

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