Switch to the permanent variable rate dropout layer in the encoding network.
Switch to the permanent variable rate dropout layer in the encoding network.

PiperOrigin-RevId: 379226308
Change-Id: I75fa0e63907bd7557bc235c85502e27ccc4c69ef

tensorflow/agents,tensorflow/agents