瀏覽代碼

update maskrcnn tutorial training configuration

will-jl944 4 年之前
父節點
當前提交
b0048dd39d
共有 1 個文件被更改,包括 1 次插入0 次删除
  1. 1 0
      dygraph/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py

+ 1 - 0
dygraph/tutorials/train/instance_segmentation/mask_rcnn_r50_fpn.py

@@ -46,6 +46,7 @@ model.train(
     train_dataset=train_dataset,
     train_dataset=train_dataset,
     train_batch_size=1,
     train_batch_size=1,
     eval_dataset=eval_dataset,
     eval_dataset=eval_dataset,
+    pretrain_weights='COCO',
     learning_rate=0.00125,
     learning_rate=0.00125,
     lr_decay_epochs=[8, 11],
     lr_decay_epochs=[8, 11],
     warmup_steps=10,
     warmup_steps=10,