Fix multi-lq bug

This commit is contained in:
James Betker 2020-05-06 23:16:35 -06:00
parent aa0305def9
commit dbca0d328c

View File

@ -14,7 +14,7 @@ class LQGTDataset(data.Dataset):
"""
def get_lq_path(self, i):
which_lq = random.randint(0, len(self.paths_LQ))
which_lq = random.randint(0, len(self.paths_LQ)-1)
return self.paths_LQ[which_lq][i]
def __init__(self, opt):