ViperEkura
/
AstrAI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix: reset sampler iter at epoch end so progress bar shows total after first epoch

This commit is contained in:
ViperEkura 2026-07-04 06:35:29 +08:00
parent 27d1921d9c
commit 27524ad085
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
astrai/dataset/sampler.py
View File

@ -74,6 +74,7 @@ class ResumableDistributedSampler(Sampler[int]):
self.epoch += 1 self.epoch += 1
self._indices = None self._indices = None
self.iter = self.iter % self.num_samples_per_replica
@property @property
def _remaining(self): def _remaining(self):