Deep Learning, Jupyter Notebook, TensorFlow, Python, Data Science.
model_training.py in the components folder. We update the same in the main.py after the prepare base model stage. STAGE_NAME = "Training" try: logger.info(f"*******************") logger.info(f">>>>>> stage {STAGE_NAME} started <<<<<<") model_trainer = ModelTrainingPipeline() model_trainer.main()