Project-DVC-CNN-CatVsDog/src/stage_03_prepare_callback.py

import argparse
import os
import logging
import time
from src.utils.common import read_yaml_file, create_directories
from src.utils.callbacks import create_and_save_tensorboard_callback,create_and_save_checkpointing_callback



logging.basicConfig(
    filename=os.path.join("logs", 'running_logs.log'), 
    level=logging.INFO, 
    format="[%(asctime)s: %(levelname)s: %(module)s]: %(message)s",
    filemode="a"
    )

def prepare_callbacks(config_path: str) -> None:
    """prepare and save callbacks as binary

    Args:
        config_path (str): path to configuration file
    """
    config = read_yaml_file(config_path)

    artifacts = config["artifacts"]
    artifacts_dir = artifacts["ARTIFACTS_DIR"]

    tensorboard_log_dir = os.path.join(artifacts_dir, artifacts["TENSORBOARD_ROOT_LOG_DIR"])

    checkpoint_dir = os.path.join(artifacts_dir, artifacts["CHECKPOINT_DIR"])

    callbacks_dir = os.path.join(artifacts_dir, artifacts["CALLBACKS_DIR"])

    create_directories([
        tensorboard_log_dir,
        checkpoint_dir,
        callbacks_dir
    ])

    create_and_save_tensorboard_callback(callbacks_dir, tensorboard_log_dir)

    create_and_save_checkpointing_callback(callbacks_dir, checkpoint_dir)





if __name__ == '__main__':
    args = argparse.ArgumentParser()
    args.add_argument("--config", "-c", default="configs/config.yaml")
    parsed_args = args.parse_args()

    try:
        logging.info("\n********************")
        logging.info(">>>>> stage three started <<<<<")
        prepare_callbacks(config_path=parsed_args.config)
        logging.info(">>>>> stage three completed! callbacks are prepared and saved as binary <<<<<n")
    except Exception as e:
        logging.exception(e)
        raise e