@@ -55,5 +55,6 @@ model.train(
warmup_start_lr=0.0,
save_interval_epochs=5,
lr_decay_epochs=[243, 324],
+ use_ema=True,
save_dir='output/ppyolo_r50vd_dcn',
use_vdl=True)
@@ -54,6 +54,7 @@ model.train(
warmup_steps=1000,
lr_decay_epochs=[430, 540, 610],
save_dir='output/ppyolotiny',
@@ -58,5 +58,6 @@ model.train(
lr_decay_epochs=[243],
save_dir='output/ppyolov2_r50vd_dcn')