Commit 90ac0229 authored by Asim Shankar's avatar Asim Shankar Committed by TensorFlower Gardener
Browse files

eager: Fix dropout in MNIST example. 

The dropout layer takes a training argument in __call__, which defaults
to false. So without this change, dropout was not being applied.

PiperOrigin-RevId: 184064465
parent 371923a0
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment