njzjz/deepmd-kit

This test for hybrid descriptor should be removed!

Closed this issue · 0 comments

Line: 120

self.file_model_param = Path(CUR_DIR) / "models" / "dpa2.pth"
self.file_type_embed = Path(CUR_DIR) / "models" / "dpa2_tebd.pth"
# TODO This test for hybrid descriptor should be removed!
def test_descriptor_hyb(self):
# torch.manual_seed(0)
model_hybrid_dpa2 = self.model_json
dparams = model_hybrid_dpa2["descriptor"]
ntypes = len(model_hybrid_dpa2["type_map"])
dlist = dparams.pop("list")
des = DescrptBlockHybrid(