OOM 报错解决

375 阅读1分钟

报错:

ResourceExhaustedError:OOM when allocating tensor with shape[9191,200] and type float on /job:localhost/replica:0/device:GPU:0 by allocatior GPU_bfc

解决:

这种情况就是参数设置太大,GPU 不够用了,只需要把参数改小,比如 batch_size,embedding_size, hidden_dim 等等,或者直接换个大的 GPU ,一劳永逸