feat: update training loop with learning rate scheduler and progress bar enhancements

feat: enhance training loop with tqdm progress bar and configurable parameters

feat: add seed setting for deterministic training

feat: update model training and logging, add mypy cache to .gitignore

feat: baseline model

feat: add init model