add remarks/todos to vgg_basic

models/vgg_basic.py

@@ -114,6 +114,12 @@ class MultiTagger(pl.LightningModule):
         return yy/frames_to_process
 
     def training_step(self, data_batch, batch_nb):
+
+        # TODO: this is not doing a backward pass?? something like this:
+        #                 self.optimizer.zero_grad()
+        #                 loss.backward()
+        #                 self.optimizer.step()
+
         x, y = data_batch
         y_hat = self.forward_full_song(x, y)
         y = y.float()
@@ -136,7 +142,7 @@ class MultiTagger(pl.LightningModule):
                 'rocauc':avg_auc}
 
     def configure_optimizers(self):
-        return [torch.optim.Adam(self.parameters(), lr=0.02)]
+        return [torch.optim.Adam(self.parameters(), lr=0.02)]  # TODO: isn't that lr too big?
 
     @pl.data_loader
     def tng_dataloader(self):