Skip to content

Commit

Permalink
fix py lint apache#2
Browse files Browse the repository at this point in the history
  • Loading branch information
khaotik committed Mar 21, 2021
1 parent 0ed7bc5 commit 90229e7
Showing 1 changed file with 7 additions and 7 deletions.
14 changes: 7 additions & 7 deletions python/mxnet/ndarray/contrib.py
Original file line number Diff line number Diff line change
Expand Up @@ -723,13 +723,13 @@ def multi_mp_adabelief_update(weights, grads, mean, var, weights32, rescale_grad
rescale_grad = _get_rescale_grad(rescale_grad, ctx=weights[0].context)
temp_list = _flatten_list(zip(weights, grads, mean, var, weights32)) + [rescale_grad]
return ndarray._internal._multi_mp_adabelief_update(*temp_list,
out=out,
num_weights=size,
lrs=lrs,
wds=wds,
etas=etas,
name=name,
**kwargs)
out=out,
num_weights=size,
lrs=lrs,
wds=wds,
etas=etas,
name=name,
**kwargs)

def multi_lans_update(weights, grads, mean, var, step_count,
lrs, wds, out=None, num_tensors=0, **kwargs):
Expand Down

0 comments on commit 90229e7

Please sign in to comment.