
import argparse
import os


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Check experiments progress')
    parser.add_argument("--input_dir", type=str, required=True)
    args = parser.parse_args()

    print('='*20)
    print(f'Mame: {args.input_dir}')

    all_dir = os.listdir(args.input_dir)
    num_exp = len(all_dir)
    cnt = 0
    for i in all_dir:
        if os.path.exists(os.path.join(args.input_dir, i, 'done')):
            cnt += 1

    print(f'[{cnt}/{num_exp}] is done')

import argparse
import os


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Check experiments progress')
    parser.add_argument("--input_dir", type=str, required=True)
    args = parser.parse_args()

    print('='*20)
    print(f'Mame: {args.input_dir}')

    all_dir = os.listdir(args.input_dir)
    num_exp = len(all_dir)
    cnt = 0
    for i in all_dir:
        if os.path.exists(os.path.join(args.input_dir, i, 'done')):
            cnt += 1

    print(f'[{cnt}/{num_exp}] is done')