Ejemplo n.º 1
0
def get_checkpoints_dir(train_dir: str):
    return get_subdir(train_dir, "checkpoints", create=True)
Ejemplo n.º 2
0
def get_train_root_dir(base_dir: str, model_name: str, create: bool):
    return get_subdir(base_dir, model_name, create)
Ejemplo n.º 3
0
def get_train_logs_dir(train_dir: str):
    return get_subdir(train_dir, "logs", create=True)
Ejemplo n.º 4
0
def get_train_dir(base_dir: str, train_name: str, create: bool):
    return get_subdir(base_dir, train_name, create)
Ejemplo n.º 5
0
def get_val_dir(train_dir: str, entry: PreparedData, iteration: int):
    subdir_name = f"{datetime.datetime.now():%Y-%m-%d,%H-%M-%S},id={entry.entry_id},speaker={entry.speaker_id},it={iteration}"
    return get_subdir(_get_validation_root_dir(train_dir),
                      subdir_name,
                      create=True)
Ejemplo n.º 6
0
def get_inference_root_dir(train_dir: str):
    return get_subdir(train_dir, "inference", create=True)
Ejemplo n.º 7
0
def _get_validation_root_dir(train_dir: str):
    return get_subdir(train_dir, "validation", create=True)
Ejemplo n.º 8
0
def get_analysis_root_dir(train_dir: str):
    return get_subdir(train_dir, "analysis", create=True)
Ejemplo n.º 9
0
def get_infer_sent_dir(infer_dir: str, result_name: str) -> str:
  return get_subdir(infer_dir, result_name, create=True)
Ejemplo n.º 10
0
def get_infer_dir(train_dir: str, run_name: str):
  return get_subdir(get_inference_root_dir(train_dir), run_name, create=True)
Ejemplo n.º 11
0
def get_val_entry_dir(val_dir: str, result_name: str) -> None:
  return get_subdir(val_dir, result_name, create=True)
Ejemplo n.º 12
0
def get_val_dir(train_dir: str, run_name: str) -> str:
  return get_subdir(_get_validation_root_dir(train_dir), run_name, create=True)