Bug fixes in TF data pipeline
Bug fixes in TF data pipeline

- Reaching the end of the dataset now triggers a validation step
- Validation step now works like NumPy validation step

tum-pbs/PhiFlow,tum-pbs/PhiFlow