SMAI Assignment 3 Task 1 Task 1 Wandb report Task 2 Task 2 Wandb report Task 3 Task 3 Wandb report Task 4 Task 4 Wandb report Task 5 Task 5 Wandb report