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):