SavedModel MNIST example: add regularization losses, and add hparams
as optional tensor-valued args. With the latter, the SavedModel can have dropout with a configurable rate. PiperOrigin-RevId: 235007707
Loading
Please sign in to comment
as optional tensor-valued args. With the latter, the SavedModel can have dropout with a configurable rate. PiperOrigin-RevId: 235007707