neuroailab / tfutils

Utilities for working with tensorflow
MIT License
25 stars 8 forks source link

HDF5 total batches vs num_steps #42

Closed qbilius closed 6 years ago

qbilius commented 7 years ago

Total batches in HDF5Provider is incorrect for the purpose of determining num_steps. It returns the number of batches for the batch size that we use to enqueue. However, the number of batches should be computed using the dequeue batch size. In the current implementation, get_validation_target does not have queue passed to it, so fixing it there is not possible. Not sure about the best way to go about it though.