njzjz/deepmd-kit

consider type embeddings for DOSFitting

Closed this issue · 0 comments

Line: 718

"exclude_types": [],
"nets": self.serialize_network(
ntypes=self.ntypes,
# TODO: consider type embeddings for DOSFitting
ndim=1,
in_dim=self.dim_descrpt + self.numb_fparam + self.numb_aparam,
out_dim=self.numb_dos,
neuron=self.n_neuron,
activation_function=self.activation_function,
resnet_dt=self.resnet_dt,
variables=self.fitting_net_variables,