我在google colab上使用roberta的问答模型来解决推文情感提取问题。
但是模型无法训练,因为我得到了一个资源耗尽错误;
查看完整错误:
ResourceExhaustedError: OOM when allocating tensor with shape[32,16,128,64] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc
[[node model/tf_roberta_model/roberta/encoder/layer_._17/