Some todos : add learning rate scheduler add proper hyperparams for Adam, LR etc. add tensorboard add scripts to download the data