文件名称：lightning_gan_zoo:使用pytorch闪电和hydra配置实现的GAN模型
文件大小：45KB
文件格式：ZIP
更新时间：2024-04-20 21:57:03
Python
闪电GAN动物园 :high_voltage: :elephant: :penguin: :panda: 使用pytorch闪电和hydra配置实现的GAN模型（正在进行中） 用法示例： python train.py +expt=dc_gan dataset=celeb_a python train.py +expt=wgan dataset=mnist python train.py +expt=wgan_gp dataset=celeb_a python train.py +expt=gan_stability_r1 dataset=celeb_a 默认情况下，将在验证步骤期间在验证集上监视FID分数。 当获得最佳FID分数时，将保存模型检查点。 生成器样本和潜在空间插值将保存到输出目录。 当前支持的型号 直流电源 威根 WGAN-GP 来自R1正则化GAN
【文件预览】：
lightning_gan_zoo-main
----core()
--------models()
--------callback_fid.py(5KB)
--------figures()
--------lightning_module.py(9KB)
--------logger.py(227B)
--------submodules()
--------utils()
----conf()
--------noise_distn()
--------figures()
--------config.yaml(1004B)
--------expt()
--------dataset()
--------lr_scheduler()
----.gitignore(87B)
----README.md(786B)
----run_network.py(2KB)