Error when running finetuning about pegasus HOT 4 CLOSED

*Resource exhausted: OOM when allocating tensor with shape[8,16,1024,1024] *
I think your GPU doesn't have enough memory to compute the models. What GPU are you using?

from pegasus.

*Resource exhausted: OOM when allocating tensor with shape[8,16,1024,1024] *
I think your GPU doesn't have enough memory to compute the models. What GPU are you using?

I think it's K80.

from pegasus.

You can also try batching your training with batch_size=1 like in the following #39

from pegasus.

Thanks I'll have a try.

from pegasus.