Function get_model_dir

pytorch-model/constants.py:34–35

(task_name, model_type, model_name)

# --- Models are saved under {task_type}_task/{model_type}/{model_name}/{epoch}.pth ---
# --- Visuals are saved under {task_type}_viz/{model_type}/{model_name}/{img_name}.png ---
def get_model_dir(task_name, model_type, model_name):
    return os.path.join(f"{task_name}_task", model_type, model_name)
def get_viz_dir(task_name, model_type, model_name):
    return os.path.join(f"{task_name}_viz", model_type, model_name)