mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Throw if dataset isn't going to work with force_multiple setting

Browse Source
This commit is contained in:
James Betker 2020-11-19 23:47:00 -07:00
parent 2b2d754d8e
commit f80acfcab6
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
codes/data/base_unsupervised_image_dataset.py
View File

@ -17,6 +17,7 @@ class BaseUnsupervisedImageDataset(data.Dataset):
self.for_eval = opt['eval'] if 'eval' in opt.keys() else False self.for_eval = opt['eval'] if 'eval' in opt.keys() else False
self.scale = opt['scale'] if not self.for_eval else 1 self.scale = opt['scale'] if not self.for_eval else 1
self.paths = opt['paths'] self.paths = opt['paths']
assert (self.target_hq_size // self.scale) % self.multiple == 0 # If we dont throw here, we get some really obscure errors.
if not isinstance(self.paths, list): if not isinstance(self.paths, list):
self.paths = [self.paths] self.paths = [self.paths]
self.weights = [1] self.weights = [1]