search

mzjb / DeepH-pack

Deep neural networks for density functional theory Hamiltonian.
GNU Lesser General Public License v3.0
219 stars 46 forks source link

How to restart training model #60

Closed fangrh closed 9 months ago

fangrh commented 9 months ago

Hi,

When I run the example of using ABACUS, everything gone well except the training is canceled by the slurm system due to the time limit.

Because the maximum time of using the GPU accelerate card is limited for one submit in our cluster. I wander if I can restart the training when next submit ?

mzjb commented 9 months ago

Please refer to resume in https://deeph-pack.readthedocs.io/en/latest/keyword/train.html#:~:text=best_state_dict.pkl.-,resume,-%3A%20The%20path%20to.

fangrh commented 9 months ago

OK, it works, thank you very much.