From 1fcd3bbf5256dccd1fd0f27c55b442251a25d668 Mon Sep 17 00:00:00 2001 From: HuguesTHOMAS Date: Thu, 29 Jul 2021 16:35:35 +0000 Subject: [PATCH] . --- train_S3DIS.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/train_S3DIS.py b/train_S3DIS.py index 984fdc6..967fc6b 100644 --- a/train_S3DIS.py +++ b/train_S3DIS.py @@ -268,8 +268,8 @@ if __name__ == '__main__': config.saving_path = sys.argv[1] # Initialize datasets - training_dataset = S3DISDataset(config, set='training', use_potentials=False) - test_dataset = S3DISDataset(config, set='validation', use_potentials=False) + training_dataset = S3DISDataset(config, set='training', use_potentials=True) + test_dataset = S3DISDataset(config, set='validation', use_potentials=True) # Initialize samplers training_sampler = S3DISSampler(training_dataset)