Skip to content
Snippets Groups Projects
Commit e7b6a5b0 authored by Aron's avatar Aron Committed by Juan M. Cruz-Martinez
Browse files

Avoid TensorFlow overhead by making one step a batch rather than an

epoch.

Avoids memory overhead by only combining up to 100 steps into one epoch,
and not changing anything when using only 1 replica (i.e. on CPU).
parent feecfc9c
No related branches found
No related tags found
1 merge request!1939Avoid idle gpu
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment