layer_pi.py 文件源码

def _parse_args(self, str_arg):
        parser = argparse.ArgumentParser(description='Python Layer Parameters Pi')
        parser.add_argument('--num_classes', default=None, type=int)
        parser.add_argument('--num_data', default=None, type=int)
        args = parser.parse_args(str_arg.split())
        return args