This repository was archived by the owner on May 1, 2025. It is now read-only.
This repository was archived by the owner on May 1, 2025. It is now read-only.
Cuda out of memory issue. #83
Description
I am trying to fine-tune the model but as the model is too large so it gets out of memory error. Any idea how to get over it?
tensorflow.python.framework.errors_impl.ResourceExhaustedError: OOM when allocating tensor with shape[246534,1280] and type float on /job:localhost/replica:0/task:0/device:GPU:0 by allocator GPU_0_bfc [Op:RandomStandardNormal]