def parse_arguments():
    parser = argparse.ArgumentParser(description='Trains model')
    parser.add_argument('-a', '--algorithm', type=str, required=True)
    parser.add_argument('--arch', type=str, default='resnet18')
    parser.add_argument('-m', '--model_dir', type=str, required=True)
    parser.add_argument('-c', '--clustering_file', type=str, required=True)
    parser.add_argument('--local_rank', type=int, default=None)
    parser.add_argument('-d', '--distributed', action='store_true')
    parser.add_argument('--dist_protocol', type=str, default='env')
    return parser.parse_args()
def parse_arguments():
    parser = argparse.ArgumentParser(description='Trains model')
    parser.add_argument('-a', '--algorithm', type=str, required=True)
    parser.add_argument('--arch', type=str, default='resnet18')
    parser.add_argument('-m', '--model_dir', type=str, required=True)
    parser.add_argument('-c', '--clustering_file', type=str, required=True)
    parser.add_argument('--local_rank', type=int, default=None)
    parser.add_argument('-d', '--distributed', action='store_true')
    parser.add_argument('--dist_protocol', type=str, default='env')
    return parser.parse_args()