Error in PyTorch-Lightning when Finetuning on VQA about vilt HOT 2 CLOSED

I've just checked that your command works on my environment => python run.py with data_root=./dataset num_gpus=1 num_nodes=1 task_finetune_vqa per_gpu_batchsize=64 load_path="weights/vilt_200k_mlm_itm.ckpt"

It seems your error is caused because the training_step method returned None, which should not be happened as "vqa" is in self.current_tasks and thus the return of forward should contain {vqa_loss: Tensor} in the returning dictionary.

Since PyTorch-lightning is a rapidly changing project, I can only guarantee my code for the specific version of PL as denoted in requirements.txt: pytorch_lightning==1.1.4.
Please re-check the versions in the requirements file match your installed versions first. And if so, please report the variables in the scope of compute_vqa for further analysis.

I strongly assume that your PL version mismatches with mine. Because pytorch_lightning/trainer/training_loop.py line 304 of PL version 1.1.4 is not closure_loss = ... but this.

from vilt.

Sorry - I had been getting an error with the earlier version of pytorch_lightning as well, and upgraded just to check if it made any difference. I identified the bug in my code that had caused the earlier error, though.

from vilt.