def test_epoch_end(self, outs):
model_name = self.hparams.config["load_path"].split("/")[-1][:-5]
import os
save_prefix = os.path.split(self.hparams.config["load_path"])[0]
print('save_prefix', save_prefix)
if self.hparams.config["loss_names"]["vqa"] > 0:
objectives.vqa_test_wrapup(outs, model_name, save_prefix)
vilt_utils.epoch_wrapup(self)
def test_epoch_end(self, outs):
model_name = self.hparams.config["load_path"].split("/")[-1][:-5]
if self.hparams.config["loss_names"]["vqa"] > 0:
objectives.vqa_test_wrapup(outs, model_name)
vilt_utils.epoch_wrapup(self)
def validation_epoch_end(self, outs):
vilt_utils.epoch_wrapup(self)
def training_epoch_end(self, outs):
vilt_utils.epoch_wrapup(self)