From 4e38de9a5b2b88ba98a41640d12a89fd37aa719b Mon Sep 17 00:00:00 2001 From: Si11ium Date: Tue, 30 Jul 2019 13:28:11 +0200 Subject: [PATCH] Default Paramter Adjustment --- main.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/main.py b/main.py index 5bb9d0b..4c58913 100644 --- a/main.py +++ b/main.py @@ -28,13 +28,13 @@ fs_root = os.path.splitdrive(sys.executable)[0] parser = argparse.ArgumentParser() default_data_dir = os.path.join(os.getcwd(), 'data') parser.add_argument('--dataset', type=str, default=default_data_dir, help='dataset path') -parser.add_argument('--npoints', type=int, default=50, help='resample points number') -parser.add_argument('--model', type=str, default='checkpoint//seg_model_custom_24.pth', help='model path') -parser.add_argument('--nepoch', type=int, default=10, help='number of epochs to train for') +parser.add_argument('--npoints', type=int, default=1024, help='resample points number') +parser.add_argument('--model', type=str, default='', help='model path') +parser.add_argument('--nepoch', type=int, default=250, help='number of epochs to train for') parser.add_argument('--outf', type=str, default='checkpoint', help='output folder') parser.add_argument('--batch_size', type=int, default=8, help='input batch size') parser.add_argument('--test_per_batches', type=int, default=1000, help='run a test batch per training batches number') -parser.add_argument('--num_workers', type=int, default=0, help='number of data loading workers') +parser.add_argument('--num_workers', type=int, default=4, help='number of data loading workers') opt = parser.parse_args() print(opt)