<filename>point_transformer/test.py
'''
Author       : ZHP
Date         : 2021-05-28 23:25:56
LastEditors  : ZHP
LastEditTime : 2021-05-29 00:06:04
FilePath     : \Projects\ptzhs\point_transformer\test.py
Description  : 项目说明
Copyright    : ZHP
'''
import sys
import os
import argparse
import models

def get_args():
    # Training settings
    parser = argparse.ArgumentParser(description='ShapePart Segmentation!')
    parser.add_argument('--task', type=str, default='PartSeg', help='Task')
    # parser.add_argument('--lr_decay', action='store_true', default=False, help='use normals')
    parser.add_argument('--lr_decay', action='store_true', help='use normals')  # 不指定
    parser.add_argument('--batch', action='store_false', help='use normals')  # 默认为
    parser.add_argument('--ab', action='store_true', help='use normals')
    parser.add_argument('--abc', action='store_true',default=True, help='use normals')
    parser.add_argument('--abcd', action='store_true',default=False, help='use normals')   

    args = parser.parse_args()
    print('主要参数配置如下：\n')
    for key, value in args.__dict__.items():
        print(f'{key:^20} : {str(value):<}')
    return args

if __name__ == "__main__":
    
    model_name = "PartSeg"
    model = getattr(models, model_name)()