m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

delete backward return list test=develop (#20294)

Browse Source
revert-20712-fix_depthwise_conv
gongweibao 6 years ago committed by GitHub
parent bc6126dd07
commit 7b9e33972a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
python/paddle/fluid/contrib/mixed_precision/decorator.py
Unescape View File

@ -135,7 +135,7 @@ class OptimizerWithMixedPrecison(object):
scaled_g = g / self._loss_scaling scaled_g = g / self._loss_scaling
scaled_params_grads.append([p, scaled_g]) scaled_params_grads.append([p, scaled_g])
return scaled_params_grads, scaled_loss return scaled_params_grads
def apply_gradients(self, scaled_params_grads): def apply_gradients(self, scaled_params_grads):
""" """
@ -194,7 +194,7 @@ class OptimizerWithMixedPrecison(object):
The scaled loss by scaling factor, the list of optimize ops, and a The scaled loss by scaling factor, the list of optimize ops, and a
list of scaled parameters and gradients. list of scaled parameters and gradients.
""" """
scaled_params_grads, scaled_loss = self.backward( scaled_params_grads = self.backward(
loss, loss,
startup_program=startup_program, startup_program=startup_program,
parameter_list=parameter_list, parameter_list=parameter_list,

Write Preview
Loading…
Cancel
Save