在Google Colab上使用TF2.0自回归模型对图像进行采样时内存不足

Question

采样功能如下：

function trigger_receive_sms($from) {
  header ('Content-Type:text/xml');
  echo ('<?xml version="1.0" encoding="UTF-8"?>');
  echo ('<Response>');
  echo ('<Message>Thank you for accepting this position.');
  echo ('</Message>');
  echo ('</Response>');
  die();
}

当我运行此代码以在Google Colab上对图像进行采样时，RAM最终变为OOM，此后内核关闭。在Colab上运行时如何避免OOM？