vijayi1

I dug into this further. When training is resumed from a non-zero epoch, RayDatasetBatcher (ludwig/data/dataset/ray.py) calls self._fetch_next_epoch() twice, once during the class init method and again in the set_epoch() method,...

@arnavgarg1 sure. can you please confirm if the following steps are correct to rebase a PR and make additional changes (if needed), To rebase my branch: git fetch upstream git...

@arnavgarg1, I followed those steps exactly and got the following error during push, (dl_venv) [vijayi@sparky01 ludwig]$ git push -u origin unet-image-utils-tests To github.com:vijayi1/ludwig.git ! [rejected] unet-image-utils-tests -> unet-image-utils-tests (non-fast-forward) error:...

@arnavgarg1, merge completed and I added a few description changes for the docs. the utils tests passed.