Google Colab Pro 中神经风格迁移任务的最大分辨率是多少?

问题描述

我正在 Google Colab Pro 上实施 NST 任务,直接从他们网站上的 Keras 示例中采用。

但是我注意到如果我不把我的内容图片的分辨率降低到一定程度,就会出现错误,说:

ResourceExhaustedError 追溯(最近一次调用 最后) 在 () 208 对于 i 在范围内(1,迭代 + 1): 209 损失,grads = compute_loss_and_grads( --> 210combination_image、base_image、style_reference_image 211) 第212话

4 帧 /usr/local/lib/python3.6/dist-packages/tensorflow/python/eager/execute.py 在 quick_execute(op_name,num_outputs,输入,attrs,ctx,name) 58 ctx.ensure_initialized() 59 张量 = pywrap_tfe.TFE_Py_Execute(ctx._handle,device_name,op_name,---> 60 个输入、属性、num_outputs) 61 除了 core._NotOkStatusException 作为 e: 62 如果名称不是 None:

ResourceExhaustedError:在分配张量时出现 OOM 形状 [3,833,2047,64] 和类型浮动 /job:localhost/replica:0/task:0/device:GPU:0 分配器 GPU_0_bfc
[[node gradient_tape/strided_slice_2/StridedSliceGrad(定义于 :178) ]] 提示:如果您想查看列表 发生 OOM 时分配的张量,添加 report_tensor_allocations_upon_oom 到当前的 Runoptions 分配信息。 [操作:__inference_compute_loss_and_grads_14209]

函数调用栈:compute_loss_and_grads

似乎 GPU 无法处理超过特定阈值的分辨率。我每次都使用试错法来达到最大分辨率。最后,我只是使用一个经验法则来节省时间:将分辨率的最高维度降低到 1024。(例如,如果内容的分辨率为 1280 x 800,我会将 1280 降低到 1024,然后让另一个尺寸自动更改,保留纵横比)。

我的问题:上述错误是什么意思?最重要的是:如何在 Google Colab Pro 上计算此类任务的最大可能分辨率?

解决方法

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

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

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