文件名称：matlab代码sqrt-BigGAN-PyTorch-TPU-Distribute:用于使用TPU训练BigGAN的分布式版本（多进程）
文件大小：2.4MB
文件格式：ZIP
更新时间：2024-06-12 15:44:59
系统开源
Matlab代码sqrt BigGAN-PyTorch-TPU-发行 我已经解决了许多问题，当前回购协议的培训过程进展顺利。 整个存储库仍在开发中，如果发现任何问题，欢迎您尝试并与我讨论。 拉动请求以解决一些问题将不胜感激，欢迎所有经验水平的开发人员。 此回购协议基于作者的官方非官方PyTorch BigGAN开发 --------------下面是它的原始说明-------------- 此仓库包含来自Andrew Brock，Jeff Donahue和Karen Simonyan的BigGAN 4-8 GPU训练的代码。 这段代码是由Andy Brock和Alex Andonian编写的。 如何使用此代码 你会需要： ，版本1.0.1 tqdm，numpy，scipy和h5py ImageNet培训集 首先，您可以选择为目标数据集准备经过预处理的HDF5版本，以实现更快的I / O。 遵循此步骤（或不执行此操作），您将需要计算FID所需的Inception时刻。 这些都可以通过修改和运行来完成 sh scripts/utils/prepare_data.sh 默认情况下，这假定您的
【文件预览】：
BigGAN-PyTorch-TPU-Distribute-master
----sync_batchnorm()
--------batchnorm_reimpl.py(2KB)
--------comm.py(4KB)
--------replicate.py(3KB)
--------__init__.py(449B)
--------unittest.py(746B)
--------batchnorm.py(15KB)
----make_hdf5.py(5KB)
----calculate_inception_moments.py(4KB)
----BigGANdeep.py(22KB)
----utils.py(49KB)
----logs()
--------process_inception_log.m(445B)
--------BigGAN_ch96_bs256x8.jsonl(9KB)
--------metalog.txt(4KB)
--------compare_IS.m(2KB)
--------process_training.m(3KB)
----sample.py(8KB)
----train.py(10KB)
----datasets.py(12KB)
----inception_utils.py(12KB)
----LICENSE(1KB)
----inception_tf13.py(5KB)
----BigGAN.py(19KB)
----imgs()
--------header_image.jpg(139KB)
--------D Singular Values.png(228KB)
--------DeepSamples.png(160KB)
--------interp_sample.jpg(790KB)
--------G Singular Values.png(231KB)
--------DogBall.png(723KB)
--------Losses.png(108KB)
--------IS_FID.png(36KB)
----losses.py(821B)
----.gitignore(53B)
----layers.py(17KB)
----train_fns.py(9KB)
----animal_hash.py(32KB)
----TFHub()
--------README.md(670B)
--------converter.py(17KB)
--------biggan_v1.py(12KB)
----README.md(12KB)
----scripts()
--------launch_BigGAN_ch64_bs256x8.sh(638B)
--------launch_BigGAN_deep.sh(671B)
--------sample_cifar_ema.sh(426B)
--------launch_SAGAN_bs128x2_ema.sh(535B)
--------launch_BigGAN_bs512x4.sh(640B)
--------sample_BigGAN_bs256x8.sh(936B)
--------launch_cifar_ema.sh(406B)
--------launch_BigGAN_bs256x8.sh(640B)
--------utils()
--------launch_BigGAN_bssmall.sh(1KB)
--------launch_SNGAN.sh(507B)