matlab代码sqrt-BigOGAN-PyTorch:BigOGAN-PyTorch

Matlab代码sqrt BigGAN-PyTorch 作者的官方非官方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 默认情况下，这假定您的ImageNet训练集已下载到此目录中的根文件夹data中，并将以128x128像素分辨率准备缓存的HDF5。 在scripts文件夹中，有多个bash脚本，它们将训练具有不同批处理大小的BigGAN。 此代码假定您无权访问完整的TPU吊舱，并因此通过使用梯度累积（对多个迷你批次的平均