blueprint_args_conflictは不要なため削除、shuffleが毎回行われる不具合修正

This commit is contained in:
u-haru
2023-03-26 03:26:55 +09:00
parent 5a3d564a30
commit a4b34a9c3c
6 changed files with 2 additions and 14 deletions

View File

@@ -437,8 +437,9 @@ class BaseDataset(torch.utils.data.Dataset):
self.replacements = {}
def set_current_epoch(self, epoch):
if not self.current_epoch == epoch:
self.shuffle_buckets()
self.current_epoch = epoch
self.shuffle_buckets()
def set_current_step(self, step):
self.current_step = step