[OPS] Bug fix for optimizers.

eaplatanios · eaplatanios · commit bdfc5b0949e7 · 2020-10-07T09:27:35.000-04:00
diff --git a/modules/api/src/main/scala/org/platanios/tensorflow/api/ops/training/optimizers/Optimizer.scala b/modules/api/src/main/scala/org/platanios/tensorflow/api/ops/training/optimizers/Optimizer.scala
@@ -182,7 +182,7 @@ trait Optimizer {
       // Create the slots needed by the variables.
       Op.initializationScope {
         VariableScope.scope(name) {
-          createSlots(variables)
+          createSlots(gradientsAndVariables.map(_._2))
         }
       }
 

Original file line number	Diff line number	Diff line change
`@@ -182,7 +182,7 @@ trait Optimizer {`
`182`	`182`	`// Create the slots needed by the variables.`
`183`	`183`	`Op.initializationScope {`
`184`	`184`	`VariableScope.scope(name) {`
`185`		`- createSlots(variables)`
	`185`	`+ createSlots(gradientsAndVariables.map(_._2))`
`186`	`186`	`}`
`187`	`187`	`}`
`188`	`188`