How can one get the intermediate output of any part of DP model

[QuantumMisaka]

As, title, sometimes we want the output in the intermediate part of DP model from evaluation of target dataset, including:

• descriptor (can be done by 'eval_descriptor'
• energy of each atom (after all the fitting net but before the sum-up)
• final feature of each atom （before which projected to the energy of each atom)

How can one get the 2nd and 3rd items under DeePMD-kit platform ?

Thanks!

[njzjz]

• energy of each atom (after all the fitting net but before the sum-up)

You can set atomic=True.

Other terms don't have an interface.