VictorMylle/Kohya-ss-sd-scripts
1
0
Fork 1
mirror of https://github.com/kohya-ss/sd-scripts.git synced 2026-04-09 06:45:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove unnecessary subset line from collate

Browse Source
This commit is contained in:
rockerBOO
2023-11-05 16:58:20 -05:00
parent e545fdfd9a
commit 9c591bdb12
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
library/train_util.py
View File

@@ -4762,10 +4762,6 @@ class collator_class:
else: else:
dataset = self.dataset dataset = self.dataset
# If we split a dataset we will get a Subset
if type(dataset) is torch.utils.data.Subset:
dataset = dataset.dataset
# set epoch and step # set epoch and step
dataset.set_current_epoch(self.current_epoch.value) dataset.set_current_epoch(self.current_epoch.value)
dataset.set_current_step(self.current_step.value) dataset.set_current_step(self.current_step.value)