今天运行了一个从GitHub上面下载的代码,奈何自己的渣渣电脑显存不够大,然后就报错了
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
然后感觉到是自己笔记本显存不够的问题,于是将fit函数中的batch_size调小即可
今天运行了一个从GitHub上面下载的代码,奈何自己的渣渣电脑显存不够大,然后就报错了
Hint: If you want to see a list of allocated tensors when OOM happens, add report_tensor_allocations_upon_oom to RunOptions for current allocation info.
然后感觉到是自己笔记本显存不够的问题,于是将fit函数中的batch_size调小即可