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)
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)