Check the other epochs: https://wandb.ai/uos_mlalp/mscoco_teacher_learning_full_data/runs/j5wd3mdm/overview?workspace=user-malbarham