Attaching attributes to optimizer only if necessary.
This cl fixed the bug where if lr is in the kwargs, opt.lr only returns a float, instead of variable. PiperOrigin-RevId: 227066569
Loading
Please sign in to comment
This cl fixed the bug where if lr is in the kwargs, opt.lr only returns a float, instead of variable. PiperOrigin-RevId: 227066569