[Fix] Fix training of VPD model (#3323)
parent
b6090a1550
commit
2b3d3d0603
|
@ -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,
|
||||
|
|
|
@ -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
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue