Error reported when I use the se_atten_v2

[919607]

Dear teachers, when I tried to use the atten_v2 model after updating deepmd-kit 2.24, the following error was displayed

dargs.dargs.ArgumentKeyError: [at location `model/descriptor`] undefined key `stripped_type_embedding` is not allowed in strict mode

My specific input file is:

{
    "_comment": " model parameters",
    "model": {
	"type_map":	["Co","Cr","Fe","Mn","Ni","V"],
	"descriptor" :{
	    "type":		"se_atten_v2",
	    "stripped_type_embedding": true,
	    "smooth_type_embdding": true,
	    "set_davg_zero": false,
	    "sel":		"auto:2.0",
	    "rcut_smth":	0.20,
	    "rcut":		6.00,
	    "neuron":		[80, 160, 320],
	    "resnet_dt":	false,
	    "axis_neuron":	16,
	    "attn":	128,
        "attn_layer":	2,
        "attn_mask":	false,
        "attn_dotr":	true,
	    "seed":		1,
	},


	"fitting_net" :{
	    "neuron":		[240,240,240],
	    "resnet_dt": 	true,
	    "seed":		1,
	    "_comment":		" that's all"
	},
	
	"_comment":	" that's all",
	"type_embedding":{
		"neuron":    [2, 4, 8],
		"resnet_dt": false,
		"seed":      1
	}
    },

    "learning_rate" :{
	"type":		"exp",
	"decay_steps":	5000,
	"start_lr":	0.001,	
	"stop_lr":	3.51e-8,
	"_comment":	"that's all"
    },

    "loss" :{
	"type":		"ener",
	"start_pref_e":	0.02,
	"limit_pref_e":	1,
	"start_pref_f":	1000,
	"limit_pref_f":	1,
	"start_pref_v":	0.02,
	"limit_pref_v":	1,
	"_comment":	" that's all"
    },

    "training" : {
	"training_data": {
	    "systems":		[
						"/public/home/dpgen/set/deepmd_sets0/training_2"
						],
	    "batch_size":	"auto",
	    "_comment":		"that's all"
	},
	"validation_data":{
	    "systems":		[
						"/public/home/dpgen/set/deepmd_sets0/validation_0"
						],
	    "batch_size":	1,
	    "numb_btch":	3,
	    "_comment":		"that's all"
	},
	"numb_steps":	1000000,
	"seed":		10,
	"disp_file":	"lcurve.out",
	"disp_freq":	2000,
	"save_freq":	2000,
	"_comment":	"that's all"
    },    

    "_comment":		"that's all"
}

[njzjz]

A correct example is given in the examples directory:

deepmd-kit/examples/water/se_atten/input.json

Lines 8 to 27 in 6cf7544

	"descriptor": {
	"type": "se_atten_v2",
	"sel": 120,
	"rcut_smth": 0.50,
	"rcut": 6.00,
	"neuron": [
	25,
	50,
	100
	],
	"resnet_dt": false,
	"axis_neuron": 16,
	"seed": 1,
	"attn": 128,
	"attn_layer": 2,
	"attn_dotr": true,
	"attn_mask": false,
	"precision": "float64",
	"_comment2": " that's all"
	},

cc @iProzd

[919607]

Thank you very much for your answer.