Deep Learning, Jupyter Notebook, TensorFlow, Python, Data Science.
update the same in the main.py file after the data ingestion stage. STAGE_NAME = "Prepare base model" try: logger.info(f"*******************") logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<") prepare_base_model = PrepareBaseModelTrainingPipeline() prepare_base_model.main()