Consume too much memory Training Failing

TheAtticusProject / cuad

CUAD (NeurIPS 2021)

https://www.atticusprojectai.org/cuad

388 stars 119 forks source link

Consume too much memory Training Failing #18

Open lalitr994 opened 1 year ago

lalitr994 commented 1 year ago

When "Creating features from dataset file at .", this code consumes too much memory (I have a 110G machine).

This makes me can not run this code. Is it possible to fix this problem? can you please guide me on what GPU specification should I use for training? @TheAtticusProject @IsCoelacanth @wangdsh

NirupamReddy2 commented 9 months ago

Hey @lalitr994, I'm facing the same issue, could you please let me know if you have a fix for this.

Thanks!