Fix AdamOptimizer usage in a defun
Adds init_scope around Adam's non-slot variable lookup so it's always accessed in a consistent context. PiperOrigin-RevId: 205335191
Loading
Please sign in to comment
Adds init_scope around Adam's non-slot variable lookup so it's always accessed in a consistent context. PiperOrigin-RevId: 205335191