minor changes

Little-Podi · Aug 8, 2022 · 2305bb8 · 2305bb8
1 parent 8ab2468
commit 2305bb8
Show file tree

Hide file tree

Showing 4 changed files with 16 additions and 16 deletions.
diff --git a/experiments/aiatrack/baseline.yaml b/experiments/aiatrack/baseline.yaml
@@ -30,33 +30,33 @@ DATA:
     SAMPLE_PER_EPOCH: 60000
 MODEL:
   BACKBONE:
-    DILATION: false
+    DILATION: False
     OUTPUT_LAYERS:
       - layer3
     TYPE: resnet50
   HEAD_TYPE: CORNER
   HIDDEN_DIM: 256
   NUM_OBJECT_QUERIES: 1
   POSITION_EMBEDDING: sine
-  PREDICT_MASK: false
+  PREDICT_MASK: False
   TRANSFORMER:
     DEC_LAYERS: 1
     DIM_FEEDFORWARD: 1024
-    DIVIDE_NORM: false
+    DIVIDE_NORM: False
     DROPOUT: 0.1
     ENC_LAYERS: 3
     NHEADS: 4
-    PRE_NORM: false
+    PRE_NORM: False
   AIA:
-    USE_AIA: true
+    USE_AIA: True
     MATCH_DIM: 64
     FEAT_SIZE: 400
 TRAIN:
   BACKBONE_MULTIPLIER: 0.1
   BATCH_SIZE: 15
-  DEEP_SUPERVISION: false
+  DEEP_SUPERVISION: False
   EPOCH: 500
-  FREEZE_BACKBONE_BN: true
+  FREEZE_BACKBONE_BN: True
   FREEZE_LAYERS:
     - conv1
     - layer1

diff --git a/experiments/aiatrack/baseline_got.yaml b/experiments/aiatrack/baseline_got.yaml
@@ -24,33 +24,33 @@ DATA:
     SAMPLE_PER_EPOCH: 60000
 MODEL:
   BACKBONE:
-    DILATION: false
+    DILATION: False
     OUTPUT_LAYERS:
       - layer3
     TYPE: resnet50
   HEAD_TYPE: CORNER
   HIDDEN_DIM: 256
   NUM_OBJECT_QUERIES: 1
   POSITION_EMBEDDING: sine
-  PREDICT_MASK: false
+  PREDICT_MASK: False
   TRANSFORMER:
     DEC_LAYERS: 1
     DIM_FEEDFORWARD: 1024
-    DIVIDE_NORM: false
+    DIVIDE_NORM: False
     DROPOUT: 0.1
     ENC_LAYERS: 3
     NHEADS: 4
-    PRE_NORM: false
+    PRE_NORM: False
   AIA:
-    USE_AIA: true
+    USE_AIA: True
     MATCH_DIM: 64
     FEAT_SIZE: 400
 TRAIN:
   BACKBONE_MULTIPLIER: 0.1
   BATCH_SIZE: 15
-  DEEP_SUPERVISION: false
+  DEEP_SUPERVISION: False
   EPOCH: 500
-  FREEZE_BACKBONE_BN: true
+  FREEZE_BACKBONE_BN: True
   FREEZE_LAYERS:
     - conv1
     - layer1

diff --git a/lib/test/analysis/plot_results.py b/lib/test/analysis/plot_results.py
@@ -313,7 +313,7 @@ def generate_formatted_report(row_labels, scores, table_name=''):
 
     for trk_id, d_name in enumerate(row_labels):
         # Display name
-        report_text = '{prev}{tracker: <{width}} |'.format(prev=report_text, tracker=d_name,
+        report_text = '{prev} {tracker: <{width}} |'.format(prev=report_text, tracker=d_name,
                                                            width=name_width)
         for (score_type, score_value), s_w in zip(scores.items(), score_widths):
             report_text = '{prev} {score: <{width}} |'.format(prev=report_text,

diff --git a/lib/train/trainers/base_trainer.py b/lib/train/trainers/base_trainer.py
@@ -61,7 +61,7 @@ def update_settings(self, settings=None):
 
             if self.settings.local_rank in [-1, 0]:
                 if not os.path.exists(self._checkpoint_dir):
-                    # print("training with multiple GPUs, checkpoints directory doesn't exist ")
+                    # print("training with multiple GPUs, checkpoints directory doesn't exist")
                     # print('create checkpoints directory ...')
                     os.makedirs(self._checkpoint_dir)
         else: