文件名称:BigGAN-TPU-TensorFlow:TPU支持Tensorflow实施“用于高保真自然图像合成的大规模GAN训练”(BigGAN)
文件大小:2.01MB
文件格式:ZIP
更新时间:2024-05-30 13:19:26
Python
BigGAN Tensorflow TPU 简单的Tensorflow TPU实施 我(David Mack)一直在修改此网络,以允许对其自我注意进行配置,以促进对不同自我注意体系结构有效性的实验。 实施说明/问题 待办事项:实现BigGAN深度架构(更简单的类嵌入,更深的resblock) 待办事项:探讨是否应使用orthogonal initialization (论文的方法)代替random normal initialization (当前实现) 待办事项:在预测和评估期间实施指数平均参数/批范数采样 待办事项:查找初始分数中的错误并实施FID 用法 建立数据 对于ImageNet,请使用来创建您选择的图像大小(例如128x128)的TFRecord文件。 --tfr-format inception 您还可以使用的数据构建脚本。 --tfr-format progan
【文件预览】:
BigGAN-TPU-TensorFlow-master
----args.py(8KB)
----launch_tpu_trace.sh(163B)
----.gitignore(1KB)
----launch_train_local_tpu_mnist.sh(291B)
----image_processing.py(12KB)
----main_tpu.py(1KB)
----launch_tensorboard.sh(102B)
----main_gpu.py(716B)
----assets()
--------128.png(251KB)
--------512.png(313KB)
--------architecture.png(211KB)
--------256.png(292KB)
--------main.png(1MB)
----LICENSE(1KB)
----.DS_Store(6KB)
----launch_train_tpu_cifar10.sh(423B)
----launch_samples.sh(70B)
----debug_input.py(3KB)
----launch_train_local_tpu_smoke_test.sh(341B)
----utils.py(6KB)
----launch_train_local_gpu_smoketest.sh(216B)
----README.md(2KB)
----launch_train_tpu_smoke_test.sh(474B)
----Pipfile(364B)
----input.py(5KB)
----BigGAN.py(12KB)
----tpu_normalization.py(5KB)
----launch_tpu_profile.sh(106B)
----inception_score.py(3KB)
----ops.py(17KB)
----main_loop.py(2KB)
----launch_train_tpu_sagan.sh(510B)
----launch_train_tpu_mnist.sh(415B)
----launch_train_tpu_sagan_sm.sh(544B)