mirrors
/
mmsegmentation
mirror of https://github.com/open-mmlab/mmsegmentation.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/3264/head
Peng Lu 2023-09-15 09:39:33 +08:00 committed by GitHub
parent b6090a1550
commit 2b3d3d0603
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
configs/vpd/vpd_sd_4xb8-25k_nyu-480x480.py
View File

@ -27,7 +27,7 @@ default_hooks = dict(checkpoint=dict(save_best='rmse', rule='less'))
# custom optimizer # custom optimizer
optim_wrapper = dict( optim_wrapper = dict(
type='ForceDefaultOptimWrapperConstructor', constructor='ForceDefaultOptimWrapperConstructor',
paramwise_cfg=dict( paramwise_cfg=dict(
bias_decay_mult=0, bias_decay_mult=0,
force_default_settings=True, force_default_settings=True,

2
mmseg/datasets/transforms/transforms.py
View File

@ -1027,7 +1027,7 @@ class RandomFlip(MMCV_RandomFlip):
for key in results.get('seg_fields', []): for key in results.get('seg_fields', []):
if results.get(key, None) is not None: if results.get(key, None) is not None:
results[key] = self._flip_seg_map( 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 results['swap_seg_labels'] = self.swap_seg_labels