search

snapthat / TF-T5-text-to-text

This repository demonstrate training T5 transformers using tensorflow 2
MIT License
12 stars 2 forks source link

Make sure that your dataset or generator can generate at least `steps_per_epoch * epochs` batches #3

Open tomacai opened 3 years ago

tomacai commented 3 years ago

Running TF-T5- Training.ipynb in colab on starting the second trainig fails with

Make sure that your dataset or generator can generate at least `steps_per_epoch * epochs` batches

steps_per_epoch is int(np.ceil(ntrain/batch_size)) Where ntrain=87599 and batch_size=4 and epochs is 5.