From 171bf40cd562e826d3045e54dbfa59e963c13832 Mon Sep 17 00:00:00 2001 From: Saransh Karira Date: Wed, 17 Oct 2018 16:50:36 +0530 Subject: [PATCH] Update train.py --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 3c3513a3..69b7756d 100644 --- a/train.py +++ b/train.py @@ -59,7 +59,7 @@ train_dataset = dataset.lmdbDataset(root=opt.trainroot) assert train_dataset -if not opt.random_sample: +if opt.random_sample: sampler = dataset.randomSequentialSampler(train_dataset, opt.batchSize) else: sampler = None