Add softmax before accuracy calculation (#408)

docs/source/tutorials_source/package/tutorial_moco_memory_bank.py

@@ -306,6 +306,7 @@ def training_epoch_end(self, outputs):
     def validation_step(self, batch, batch_idx):
         x, y, _ = batch
         y_hat = self.forward(x)
+        y_hat = torch.nn.functional.softmax(y_hat, dim=1)
         self.accuracy(y_hat, y)
         self.log('val_acc', self.accuracy.compute(),
                  on_epoch=True, prog_bar=True)