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

fix nasnet GradOperation

This commit is contained in:
panfengfeng 2020-10-15 20:11:33 +08:00
parent fa37f8fde3
commit 5537ec8d92
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
model_zoo/official/cv/nasnet/src/nasnet_a_mobile.py
View File

@ -911,7 +911,7 @@ class NASNetAMobileTrainOneStepWithClipGradient(nn.Cell):
self.network.add_flags(defer_inline=True) self.network.add_flags(defer_inline=True)
self.weights = optimizer.parameters self.weights = optimizer.parameters
self.optimizer = optimizer self.optimizer = optimizer
self.grad = C.GradOperation('grad', get_by_list=True, sens_param=True) self.grad = C.GradOperation(get_by_list=True, sens_param=True)
self.hyper_map = C.HyperMap() self.hyper_map = C.HyperMap()
self.sens = sens self.sens = sens
self.reducer_flag = False self.reducer_flag = False