OOM when allocating tensor

1、减小batchsize
2、采用cpu训练
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "-1"

posted @ 2021-10-21 13:56  轩辕吊雷  阅读(45)  评论(0)    收藏  举报