deepmodeling / DeePTB

DeePTB: A deep learning package for tight-binding approach with ab initio accuracy.
GNU Lesser General Public License v3.0
54 stars 16 forks source link

fix: the error occur when loading checkpoint from cuda to cpu or to cuda to continuous training #182

Closed floatingCatty closed 5 months ago