ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[32,128,128,128]
出现这种错误的原因是超出内存了,因此可以适当减小batch_size的大小即可解决,或者使用generator
ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[32,128,128,128]
出现这种错误的原因是超出内存了,因此可以适当减小batch_size的大小即可解决,或者使用generator