Skip to content

Commit 0546857

Browse files
author
Yurii Shevchuk
committed
removed explicit float cast for adam algorithm
1 parent a6501b7 commit 0546857

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

neupy/algorithms/gd/adam.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -85,10 +85,10 @@ def init_param_updates(self, layer, parameter):
8585
value=asfloat(np.zeros(parameter_shape)),
8686
)
8787

88-
step = asfloat(self.variables.step)
89-
beta1 = asfloat(self.beta1)
90-
beta2 = asfloat(self.beta2)
91-
epsilon = asfloat(self.epsilon)
88+
step = self.variables.step
89+
beta1 = self.beta1
90+
beta2 = self.beta2
91+
epsilon = self.epsilon
9292

9393
gradient = T.grad(self.variables.error_func, wrt=parameter)
9494

0 commit comments

Comments
 (0)