Throw if dataset isn't going to work with force_multiple setting
This commit is contained in:
parent
2b2d754d8e
commit
f80acfcab6
|
@ -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]
|
||||||
|
|
Loading…
Reference in New Issue
Block a user