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