CUDA out of memory - Githubissues

THU-BPM / SelfORE

The source code of paper "SelfORE: Self-supervised Relational Feature Learning for Open Relation Extraction"

47 stars 10 forks source link

Closed h-peng17 closed 4 years ago

h-peng17 commented 4 years ago

The model always exceeds gpu memory even i set batch_size to 1. So I wonder what's the size of your gpu? Thank you

xuminghu commented 4 years ago

Our GPU machine has a total of 7 TITAN RTX (24G each), the SelfORE code needs a similar amount of resources to be run.

h-peng17 commented 4 years ago

Thank you!