yankf21
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix model zoo fossid check failed

This commit is contained in:
caojian05 2021-07-02 17:18:03 +08:00
parent e1d25946bc
commit 39b42238ff
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