- code needs to be updated to run with our dataset (we don't have all the hardcoded dataset). - track on wandb - loading saved models on wandb