copy variables into state params for type casting

1 files changed, 2 insertions, 2 deletions

diff --git a/sgd.lua b/sgd.lua
index e362d6c..8ad59e4 100644
--- a/sgd.lua
+++ b/sgd.lua
@@ -49,7 +49,7 @@ function optim.sgd(opfunc, x, config, state)
       if not state.decayParameters then
          state.decayParameters = torch.Tensor():typeAs(x):resizeAs(dfdx)
       end
-      state.decayParameters:cmul(wds, x)
+      state.decayParameters:copy(wds):cmul(x)
       dfdx:add(state.decayParameters)
    end
 
@@ -75,7 +75,7 @@ function optim.sgd(opfunc, x, config, state)
       if not state.deltaParameters then
          state.deltaParameters = torch.Tensor():typeAs(x):resizeAs(dfdx)
       end
-      state.deltaParameters:cmul(lrs, dfdx)
+      state.deltaParameters:copy(lrs):cmul(dfdx)
       x:add(-clr, state.deltaParameters)
    else
       x:add(-clr, dfdx)