Commit e4b4b2e4 authored by Priya Gupta's avatar Priya Gupta Committed by TensorFlower Gardener
Browse files

Build different models for training, test and predict in keras when used with... 

Build different models for training, test and predict in keras when used with distribution strategy.

NOTE: This is already the case when used with TPUStrategy. This just extends this to all other strategies. This change enables us to run at higher batch sizes with running into OOM issues.
PiperOrigin-RevId: 231129071
parent c1162e86
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