Function parse_args

OpenSelfSup/tools/count_parameters.py:7–11

()

def parse_args():
    parser = argparse.ArgumentParser(description='Train a model')
    parser.add_argument('config', help='train config file path')
    args = parser.parse_args()
    return args


def main():