error: cuda out of memory. when training a YIG model

mzjb / DeepH-pack

Deep neural networks for density functional theory Hamiltonian.

GNU Lesser General Public License v3.0

246 stars 51 forks source link

error: cuda out of memory. when training a YIG model #79

Open Lonya0 opened 2 months ago

Lonya0 commented 2 months ago

Using nvidia rtx 4090 (24G memory), to train a YIG model, whose hamiltonian.h5 file sizes 230MB, encountered error: cuda out of memory. I have go through CrI3 test and successed before. How can I make it possible to train this model? 1725765332852

mzjb commented 3 weeks ago

You could try switching to DeepH-E3 for training. It has the same data interface as DeepH but requires less GPU memory.