ss-Res101FPN

Ranchentx · Sep 18, 2019 · 5e6b2be · 5e6b2be
1 parent d546b87
commit 5e6b2be
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/fcos/fcos_R_101_FPN_ss_2x.yaml b/configs/fcos/fcos_R_101_FPN_ss_2x.yaml
@@ -24,5 +24,5 @@ SOLVER:
   WEIGHT_DECAY: 0.0001
   STEPS: (120000, 160000)
   MAX_ITER: 180000
-  IMS_PER_BATCH: 4
+  IMS_PER_BATCH: 16
   WARMUP_METHOD: "constant"
diff --git a/tools/train_net.py b/tools/train_net.py
@@ -9,7 +9,7 @@
 
 import argparse
 import os
-os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
+# os.environ["CUDA_VISIBLE_DEVICES"] = "4,5,6,7"
 
 import torch
 from fcos_core.config import cfg