BigGAN-PyTorch icon indicating copy to clipboard operation
BigGAN-PyTorch copied to clipboard

Fix MultiEpochSampler __len__() > 0 error

Open JonFreer opened this issue 3 years ago • 1 comments

Fix for Issue #71

Caused when len(self.data_source) * self.num_epochs < self.start_itr * self.batch_size.

This happens once an entire pass of the dataset has occurred and then you resume training.

Fix: sets start_itr to the relative offset, resets to 0 after run through the dataset.

JonFreer avatar Nov 17 '20 19:11 JonFreer

thanks

szulm avatar Dec 19 '21 05:12 szulm