tensorflow / tensorflow

An Open Source Machine Learning Framework for Everyone

Home Page:https://tensorflow.org

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Fit function to work with adaptive step size

Danojungo opened this issue · comments

Hi,
Is there a way to make the generic fit function work but have different steps in each epoch?
I'm using tf.dataset.sample_from_datasets and I don't get the same amount of steps each epoch, I don't want to use repeat and I do want to run over all my positive samples without setting the steps_per_epoch.