attgan设置batch_size=1 infer报错
Created by: JiaXiao243
在 PaddleGAN目录下执行python infer.py --batch_size 1 --model_net AttGAN --init_model output_attgan/checkpoints/40/ --dataset_dir "data/celeba/" --image_size 128 --output infer_attgan,对该模型进行预测报错信息如下:
infer的batch_size设置为其他值,可以正常infer。