Ge Zheng
commited on
Commit
·
d4c3ee7
1
Parent(s):
31c55f2
Reduce warmup epochs from 5 to 1 when finetuning on custom data. (#418)
Browse files
exps/example/yolox_voc/yolox_voc_s.py
CHANGED
@@ -14,6 +14,7 @@ class Exp(MyExp):
|
|
14 |
self.num_classes = 20
|
15 |
self.depth = 0.33
|
16 |
self.width = 0.50
|
|
|
17 |
self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
|
18 |
|
19 |
def get_data_loader(self, batch_size, is_distributed, no_aug=False, cache_img=False):
|
|
|
14 |
self.num_classes = 20
|
15 |
self.depth = 0.33
|
16 |
self.width = 0.50
|
17 |
+
self.warmup_epochs = 1
|
18 |
self.exp_name = os.path.split(os.path.realpath(__file__))[1].split(".")[0]
|
19 |
|
20 |
def get_data_loader(self, batch_size, is_distributed, no_aug=False, cache_img=False):
|