[Fix] Fix training of VPD model (#3323)

open-mmlab · Sep 15, 2023 · 2b3d3d0 · 2b3d3d0
1 parent b6090a1
commit 2b3d3d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/configs/vpd/vpd_sd_4xb8-25k_nyu-480x480.py b/configs/vpd/vpd_sd_4xb8-25k_nyu-480x480.py
@@ -27,7 +27,7 @@
 
 # custom optimizer
 optim_wrapper = dict(
-    type='ForceDefaultOptimWrapperConstructor',
+    constructor='ForceDefaultOptimWrapperConstructor',
     paramwise_cfg=dict(
         bias_decay_mult=0,
         force_default_settings=True,

diff --git a/mmseg/datasets/transforms/transforms.py b/mmseg/datasets/transforms/transforms.py
@@ -1027,7 +1027,7 @@ def _flip(self, results: dict) -> None:
         for key in results.get('seg_fields', []):
             if results.get(key, None) is not None:
                 results[key] = self._flip_seg_map(
-                    results[key], direction=results['flip_direction'])
+                    results[key], direction=results['flip_direction']).copy()
                 results['swap_seg_labels'] = self.swap_seg_labels