Exemple #1
0
    print(f'Test - Input Eval: {print_eval_xs}')
    print(f'Test - Out Eval: {print_eval_outs}')

    path_eval = Path(hp.logdir, f'test_{test_epoch}', 'test_eval.txt')
    if not path_eval.exists():
        print_to_file(path_eval, print_eval, (print_eval_xs, print_eval_outs))

    runner.writer.close()


if __name__ == '__main__':
    parser = ArgumentParser()
    parser.add_argument('--test', type=int, default=-1)

    args = hp.parse_argument(parser)
    test_epoch = args.test
    if test_epoch == -1:
        # check overwrite or not
        if list(Path(hp.logdir).glob('events.out.tfevents.*')):
            while True:
                s = input(
                    f'"{hp.logdir}" already has tfevents. continue? (y/n)\n')
                if s.lower() == 'y':
                    shutil.rmtree(hp.logdir)
                    os.makedirs(hp.logdir)
                    break
                elif s.lower() == 'n':
                    exit()

    main(test_epoch)
Exemple #2
0

if __name__ == '__main__':
    # determined by sys argv
    parser = ArgumentParser()
    parser.add_argument('room_create')
    parser.add_argument('kind_data',
                        choices=('TRAIN', 'train',
                                 'SEEN', 'seen',
                                 'UNSEEN', 'unseen',
                                 ),
                        )
    parser.add_argument('-t', dest='target_folder', default='')
    parser.add_argument('--from', type=int, default=-1,
                        dest='from_idx')
    args = hp.parse_argument(parser, print_argument=False)
    use_dirac = hp.feature == 'DirAC'
    n_cuda_dev = len(hp.device)
    is_train = args.kind_data.lower() == 'train'

    # Paths
    path_speech = hp.dict_path['speech_train' if is_train else 'speech_test']

    if args.target_folder:
        path_result = hp.path_feature / args.target_folder
        if not is_train:
            path_result = path_result / 'TEST'
        path_result = path_result / args.kind_data.upper()
    else:
        path_result = hp.dict_path[f'feature_{args.kind_data.lower()}']
    os.makedirs(path_result, exist_ok=True)