StarGAN-Keras
StarGAN-Keras copied to clipboard
Incorrect arguments in test function
Hi,
Line 279 in StarGAN.py reads:
data_iter = get_loader(self.Image_data_class.test_dataset, self.Image_data_class.train_dataset_label, self.Image_data_class.train_dataset_fix_label, image_size=self.image_size, batch_size=self.batch_size, mode=self.mode)
Since this is within the test() function, shouldn't arguments 2 and 3 be "self.Image_data_class.test_dataset_label" and "self.Image_data_class.test_dataset_fix_label" ?
Ok it makes sense. I've just fixed it, thank you.