descriptor type choices

[Feiyang472]

In the training file what are possible choices for model->descriptors->type other than "loc_frame"?

Sorry if this has been brought up. I just couldn't find it in any of the documentation.

[njzjz]

deepmd-kit/source/train/Trainer.py

Lines 76 to 86 in 984ef69

	descrpt_type = j_must_have(descrpt_param, 'type')
	if descrpt_type == 'loc_frame':
	self.descrpt = DescrptLocFrame(descrpt_param)
	elif descrpt_type == 'se_a' :
	self.descrpt = DescrptSeA(descrpt_param)
	elif descrpt_type == 'se_r' :
	self.descrpt = DescrptSeR(descrpt_param)
	elif descrpt_type == 'se_ar' :
	self.descrpt = DescrptSeAR(descrpt_param)
	else :
	raise RuntimeError('unknow model type ' + descrpt_type)

Four types are supported: loc_frame, se_a, se_r, and se_ar. The last 3 types are based on the DeePot-SE method.

[Feiyang472]

Thank you for your reply.