paddle-gan 在 Windows 上运行 StarGAN V2 算法出现错误
(paddlegan) PS F:\PaddleGAN> Python -u tools/main.py --config-file configs/starganv2_celeba_hq.yaml [11/21 11:29:27] ppgan INFO: Configs: dataset: test: batch_size: 16 dataroot: data/stargan-v2/celeba_hq/val/ is_train: false name: StarGANv2Dataset num_workers: 8 preprocess: - key: src name: LoadImageFromFile - key: ref name: LoadImageFromFile - input_keys: - src - ref name: Transforms pipeline: - interpolation: bicubic keys: - image - image name: Resize size: - 256 - 256 - keys: - image - image name: Transpose - keys: - image - image mean: - 127.5 - 127.5 - 127.5 name: Normalize std: - 127.5 - 127.5 - 127.5 test_count: 16 train: batch_size: 4 dataroot: data/stargan-v2/celeba_hq/train/ is_train: true name: StarGANv2Dataset num_workers: 8 preprocess: - key: src name: LoadImageFromFile - key: ref name: LoadImageFromFile - key: ref2 name: LoadImageFromFile - input_keys: - src - ref - ref2 name: Transforms pipeline: - interpolation: bilinear keys: - image - image - image name: RandomResizedCropProb prob: 0.9 ratio: - 0.9 - 1.1 scale: - 0.8 - 1.0 size: - 256 - 256 - interpolation: bilinear keys: - image - image - image name: Resize size: - 256 - 256 - keys: - image - image - image name: RandomHorizontalFlip prob: 0.5 - keys: - image - image - image name: Transpose - keys: - image - image - image mean: - 127.5 - 127.5 - 127.5 name: Normalize std: - 127.5 - 127.5 - 127.5 epochs: 200 is_train: true log_config: interval: 100 visualiual_interval: 3000 lr_scheduler: decay_epochs: 100 iters_per_epoch: 365 learning_rate: 0.0001 name: LinearDecay start_epoch: 100 model: discriminator: img_size: 256 name: StarGANv2Discriminator num_domains: 2 fan: fname_pretrained: null name: FAN generator: img_size: 256 name: StarGANv2Generator style_dim: 64 w_hpf: 1 lambda_cyc: 1 lambda_ds: 1 lambda_sty: 1 latent_dim: 16 mapping: latent_dim: 16 name: StarGANv2Mapping num_domains: 2 style_dim: 64 name: StarGANv2Style num_domains: 2 style_dim: 64 optimizer: discriminator: beta1: 0.0 beta2: 0.99 name: Adam net_names: - discriminator weight_decay: 0.0001 generator: beta1: 0.0 beta2: 0.99 name: Adam net_names: - generator
内容来源: PaddlePaddle/PaddleGAN