Hi, I had an error while feeding male2female test images (178x218) into test.py or test_batch.py. ```bash RuntimeError: The size of tensor a (313) must match the size of tensor b (312) at non-singleton dimension 2 ``` It seems that the aspect ratio is unintentionally kept during resizing, leading to (256, 313) output instead of (256, 256). One way to fix this is changing `new_size=config['new_size']` to `new_size=(config['new_size'], config['new_size'])` in the below line. https://github.com/hyperplane-lab/ACL-GAN/blob/7166a866e7b2eb011ace06d59ceafab905075108/test_batch.py#L65