tensorflow - 减少输入神经网络的数据大小？

Question

我正在尝试来自互联网的 tensorflow 代码，
它是关于 GAN 算法的，但我遇到了一个与我机器上的 GPU 内存相关的错误，
这是我正在使用的配置

tf_config['graph_options.place_pruned_graph']   = True      
tf_config['gpu_options.allow_growth']          = True    
env.CUDA_VISIBLE_DEVICES                       = '0,1,2,3,4,5,6,7,'       
env.TF_CPP_MIN_LOG_LEVEL                        = '1'       

desc += '-preset-v2-8gpus'; num_gpus = 8; sched.minibatch_base = 32; sched.minibatch_dict = 
{4: 512, 8: 256, 16: 128, 32: 64, 64: 32}; sched.G_lrate_dict = {128: 0.0015, 256: 0.002, 
512: 0.003, 1024: 0.003}; sched.D_lrate_dict = EasyDict(sched.G_lrate_dict); 
train.total_kimg = 12000


desc += '-fp32'; sched.max_minibatch_per_gpu = {256: 16, 512: 8, 1024: 4}

我应该在哪里编辑以减少数据大小..告知更多细节......提前谢谢

标签： tensorflowgenerative-adversarial-networkbatchsize