减少馈入神经网络的数据大小？

问题描述

我正在尝试来自互联网的 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}

我应该在哪里编辑以减少数据大小..通知更多详细信息...提前致谢

解决方法

暂无找到可以解决该程序问题的有效方法，小编努力寻找整理中！

如果你已经找到好的解决方法，欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@）