我正在尝试来自互联网的 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}
我应该在哪里编辑以减少数据大小..告知更多细节......提前谢谢