[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

View File

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

View File

@ -1027,7 +1027,7 @@ class RandomFlip(MMCV_RandomFlip):
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