fix coco

shuangw98 · Jul 25, 2022 · 7c55246 · 7c55246
1 parent 50e25ea
commit 7c55246
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/coco_train.sh b/coco_train.sh
@@ -4,9 +4,9 @@ IMAGENET_PRETRAIN=weight/R-101.pkl
 IMAGENET_PRETRAIN_TORCH=weight/resnet101-5d3b4d8f.pth  
 
 
-# python3 train_net.py --num-gpus 4 --config-file configs/coco/base.yaml     \
-#     --opts MODEL.WEIGHTS ${IMAGENET_PRETRAIN}                                         \
-#            OUTPUT_DIR ${SAVEDIR}/base
+python3 train_net.py --num-gpus 4 --config-file configs/coco/base.yaml     \
+    --opts MODEL.WEIGHTS ${IMAGENET_PRETRAIN}                                         \
+           OUTPUT_DIR ${SAVEDIR}/base
 
 
 python3 tools/model_surgery.py --dataset coco --method randinit                        \