Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Number of epochs unused in dataset_utils.create_input_pipeline #79

Open
indraastra opened this issue Jul 8, 2017 · 1 comment
Open

Comments

@indraastra
Copy link
Contributor

The n_epochs variable goes unused in dataset_utils.create_input_pipeline(), so the line

    producer = tf.train.string_input_producer(
        files, capacity=len(files))

should probably be

    producer = tf.train.string_input_producer(
        files, num_epochs=n_epochs, capacity=len(files))

However, I also had to add a call to sess.run(tf.local_variables_initializer()) after making that change.

@pkmital
Copy link
Owner

pkmital commented Jul 8, 2017

Good catch thanks filing an issue!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants