随笔分类 -  tensorflow

OOM when allocating tensor
摘要：1、减小batchsize 2、采用cpu训练 import os os.environ["CUDA_VISIBLE_DEVICES"] = "-1" 阅读全文
posted @ 2021-10-21 13:56 轩辕吊雷 阅读(45) 评论(0) 推荐(0)