steffen/Labeling-Primitives-With-Point2Net
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Default Paramter Adjustment

Browse Source
This commit is contained in:
Si11ium 2019-07-30 13:28:11 +02:00
parent f2d6876591
commit 4e38de9a5b
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -28,13 +28,13 @@ fs_root = os.path.splitdrive(sys.executable)[0]
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
default_data_dir = os.path.join(os.getcwd(), 'data') 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('--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('--npoints', type=int, default=1024, help='resample points number')
parser.add_argument('--model', type=str, default='checkpoint//seg_model_custom_24.pth', help='model path') parser.add_argument('--model', type=str, default='', help='model path')
parser.add_argument('--nepoch', type=int, default=10, help='number of epochs to train for') 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('--outf', type=str, default='checkpoint', help='output folder')
parser.add_argument('--batch_size', type=int, default=8, help='input batch size') 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('--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() opt = parser.parse_args()
print(opt) print(opt)