ducnhan0804 commited on
Commit
0733fe6
·
verified ·
1 Parent(s): c5388a3

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -139,9 +139,9 @@ model:
139
  backbone: HGNetv2-B5
140
 
141
  training:
142
- total_epochs: 80
143
- stage1_epochs: 72 (heavy augmentation)
144
- stage2_epochs: 8 (fine-tuning, no heavy augmentation)
145
  resolution: 640
146
  device: cuda
147
 
 
139
  backbone: HGNetv2-B5
140
 
141
  training:
142
+ total_epochs: 35
143
+ stage1_epochs: 30 (heavy augmentation)
144
+ stage2_epochs: 5 (fine-tuning, no heavy augmentation)
145
  resolution: 640
146
  device: cuda
147