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