10 2021 档案

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

点击右上角即可分享
微信分享提示