Fix TypeError in main.py

remove extraneous argument outputs (L403, main.py)

Author: silky1708

main.py

@@ -400,7 +400,7 @@ def on_train_epoch_start(self, trainer, pl_module):
         torch.cuda.synchronize(trainer.root_gpu)
         self.start_time = time.time()
 
-    def on_train_epoch_end(self, trainer, pl_module, outputs):
+    def on_train_epoch_end(self, trainer, pl_module):
         torch.cuda.synchronize(trainer.root_gpu)
         max_memory = torch.cuda.max_memory_allocated(trainer.root_gpu) / 2 ** 20
         epoch_time = time.time() - self.start_time