def test_image_loader(self):
"""
Test that model correctly handles text task.
"""
opt = ParlaiParser().parse_args([])
opt.update(BASE_IMAGE_ARGS)
for image_mode, dim in IMAGE_MODE_TO_DIM.items():
opt["image_mode"] = image_mode
teacher = create_task_agent_from_taskname(opt)[0]
teacher_act = teacher.get(0)
self.assertEquals(
teacher_act["image"].size(),
dim,
f"dim mismatch for image mode {image_mode}",
)
def _base_test_loader(self, image_mode_partial: str, no_cuda: bool = False):
"""
Test for given partial image mode.
"""
opt = ParlaiParser().parse_args([])
opt.update(BASE_IMAGE_ARGS)
opt['no_cuda'] = no_cuda
for image_mode, dim in IMAGE_MODE_TO_DIM.items():
if image_mode_partial not in image_mode:
continue
opt["image_mode"] = image_mode
teacher = create_task_agent_from_taskname(opt)[0]
teacher_act = teacher.get(0)
self.assertEquals(
teacher_act["image"].size(),
dim,
f"dim mismatch for image mode {image_mode}",
)
torch.cuda.empty_cache()
