typo fix

YangLiu14 · Dec 21, 2019 · ff35f8e · ff35f8e
1 parent ea969c4
commit ff35f8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train_scannet.py b/train_scannet.py
@@ -74,7 +74,7 @@ def main(args):
     # dataset_test = ScannetDataset(args.data_path, get_transform(train=False), data_split='test')
 
     # Choose subset of all validation images
-    indices = torch.randperm(len(dataset)).tolist()
+    indices = torch.randperm(len(dataset_valid)).tolist()
     dataset_valid = torch.utils.data.Subset(dataset_valid, indices[:-100])
 
     # # split the dataset in train and test set