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

This commit is contained in:
James Betker 2020-11-19 23:47:00 -07:00
parent 2b2d754d8e
commit f80acfcab6

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]