-
matlab代码sqrt-biggan_vase是针对另一个回购对BigGAN进行少量返工后的版本
资源介绍
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吊舱,并因此通过使用梯度累积(对多个迷你批次的平均