Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!19296 mode zoo code refactoring 

Merge pull request !19296 from caojian05/ms_master_fossid_fix
This commit is contained in:
i-robot 2021-07-05 06:18:42 +00:00 committed by Gitee
parent a65fca39f7 39b42238ff
commit 6289da72d0
4 changed files with 28 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
model_zoo/official/cv/FCN8s/src/utils/lr_scheduler.py
View File

@ -436,15 +436,13 @@ class CyclicLR(_LRScheduler):
self.gamma = gamma
if scale_fn is None:
if self.mode == 'triangular':
self.scale_fn = self._triangular_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'triangular2':
self.scale_fn = self._triangular2_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'exp_range':
self.scale_fn = self._exp_range_scale_fn
self.scale_mode = 'iterations'
mode_map = {
'triangular': ['cycle', self._triangular_scale_fn],
'triangular2': ['cycle', self._triangular2_scale_fn],
'exp_range': ['iterations', self._exp_range_scale_fn]
}
self.scale_mode = mode_map.get(self.mode)[0]
self.scale_fn = mode_map.get(self.mode)[1]
else:
self.scale_fn = scale_fn
self.scale_mode = scale_mode

16
model_zoo/official/cv/centerface/src/lr_scheduler.py
View File

@ -484,15 +484,13 @@ class CyclicLR(_LRScheduler):
self.gamma = gamma
if scale_fn is None:
if self.mode == 'triangular':
self.scale_fn = self._triangular_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'triangular2':
self.scale_fn = self._triangular2_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'exp_range':
self.scale_fn = self._exp_range_scale_fn
self.scale_mode = 'iterations'
mode_map = {
'triangular': ['cycle', self._triangular_scale_fn],
'triangular2': ['cycle', self._triangular2_scale_fn],
'exp_range': ['iterations', self._exp_range_scale_fn]
}
self.scale_mode = mode_map.get(self.mode)[0]
self.scale_fn = mode_map.get(self.mode)[1]
else:
self.scale_fn = scale_fn
self.scale_mode = scale_mode

16
model_zoo/official/cv/densenet/src/lr_scheduler/lr_scheduler.py
View File

@ -436,15 +436,13 @@ class CyclicLR(_LRScheduler):
self.gamma = gamma
if scale_fn is None:
if self.mode == 'triangular':
self.scale_fn = self._triangular_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'triangular2':
self.scale_fn = self._triangular2_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'exp_range':
self.scale_fn = self._exp_range_scale_fn
self.scale_mode = 'iterations'
mode_map = {
'triangular': ['cycle', self._triangular_scale_fn],
'triangular2': ['cycle', self._triangular2_scale_fn],
'exp_range': ['iterations', self._exp_range_scale_fn]
}
self.scale_mode = mode_map.get(self.mode)[0]
self.scale_fn = mode_map.get(self.mode)[1]
else:
self.scale_fn = scale_fn
self.scale_mode = scale_mode

16
model_zoo/research/nlp/dscnn/src/lr_scheduler.py
View File

@ -485,15 +485,13 @@ class CyclicLR(_LRScheduler):
self.gamma = gamma
if scale_fn is None:
if self.mode == 'triangular':
self.scale_fn = self._triangular_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'triangular2':
self.scale_fn = self._triangular2_scale_fn
self.scale_mode = 'cycle'
elif self.mode == 'exp_range':
self.scale_fn = self._exp_range_scale_fn
self.scale_mode = 'iterations'
mode_map = {
'triangular': ['cycle', self._triangular_scale_fn],
'triangular2': ['cycle', self._triangular2_scale_fn],
'exp_range': ['iterations', self._exp_range_scale_fn]
}
self.scale_mode = mode_map.get(self.mode)[0]
self.scale_fn = mode_map.get(self.mode)[1]
else:
self.scale_fn = scale_fn
self.scale_mode = scale_mode