Aquila-Consortium/map2map
32
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge remote-tracking branch 'refs/remotes/origin/lag2eul' into lag2eul

Browse source
This commit is contained in:
Yin Li 2020-07-22 02:38:45 -04:00
commit f98b0fbbd0
1 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
map2map/data/sampler.py
View file

@ -38,11 +38,11 @@ class DistFieldSampler(Sampler):
self.div_shuffle_dist = div_shuffle_dist self.div_shuffle_dist = div_shuffle_dist
def __iter__(self): def __iter__(self):
# deterministically shuffle based on epoch
g = torch.Generator()
g.manual_seed(self.epoch)
if self.shuffle: if self.shuffle:
# deterministically shuffle based on epoch
g = torch.Generator()
g.manual_seed(self.epoch)
if self.div_data: if self.div_data:
# shuffle files # shuffle files
ind = torch.randperm(self.nfile, generator=g) ind = torch.randperm(self.nfile, generator=g)