mirror of
https://github.com/kohya-ss/sd-scripts.git
synced 2026-04-08 22:35:09 +00:00
blueprint_args_conflictは不要なため削除、shuffleが毎回行われる不具合修正
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user