From dbca0d328c5e23c7744f67839a6b5f00d22bcc1c Mon Sep 17 00:00:00 2001
From: James Betker <jbetker@gmail.com>
Date: Wed, 6 May 2020 23:16:35 -0600
Subject: [PATCH] Fix multi-lq bug
---
codes/data/LQGT_dataset.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/codes/data/LQGT_dataset.py b/codes/data/LQGT_dataset.py
index 2d54cda7..02bfdd74 100644
--- a/codes/data/LQGT_dataset.py
+++ b/codes/data/LQGT_dataset.py
@@ -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):