Epoch 0: 100%|##########################################################################| 4/4 [02:34<00:00, 38.51s/it, loss=0.012, v_num=34, train/loss_simple_step=0.0311, train/loss_vlb_step=0.000133, train/loss_step=0.0311, global_step=2.000Traceback (most recent call last):
File "main_inpainting.py", line 707, in
trainer.fit(model, data)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 741, in fit
self._fit_impl, model, train_dataloaders, val_dataloaders, datamodule, ckpt_path
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 685, in _call_and_handle_interrupt
return trainer_fn(*args, **kwargs)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 777, in _fit_impl
self._run(model, ckpt_path=ckpt_path)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1199, in _run
self._dispatch()
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1279, in _dispatch
self.training_type_plugin.start_training(self)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/plugins/training_type/training_type_plugin.py", line 202, in start_training
self._results = trainer.run_stage()
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1289, in run_stage
return self._run_train()
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1319, in _run_train
self.fit_loop.run()
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/loops/base.py", line 145, in run
self.advance(*args, **kwargs)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/loops/fit_loop.py", line 234, in advance
self.epoch_loop.run(data_fetcher)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/loops/base.py", line 151, in run
output = self.on_run_end()
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/loops/epoch/training_epoch_loop.py", line 298, in on_run_end
self.trainer.call_hook("on_train_epoch_end")
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/trainer.py", line 1495, in call_hook
callback_fx(*args, **kwargs)
File "/root/miniconda3/envs/ldm/lib/python3.6/site-packages/pytorch_lightning/trainer/callback_hook.py", line 93, in on_train_epoch_end
callback.on_train_epoch_end(self, self.lightning_module)
TypeError: on_train_epoch_end() missing 1 required positional argument: 'outputs'
None
May I ask if you have resolved this issue in the future？:)
lorenzo-stacchio
