How to use the v2 to train IS2RE datasets?

[mumu-1010]

As the title states，How to use the v2 to train IS2RE datasets? I think I need some tutorials.

[yilunliao]

Hi @mumu-1010

The current codebase supports only S2EF.
To use IS2RE, you can use the same energy prediction head for relaxed energy and remove the force prediction head.
You can also use the force prediction head for IS2RS as we did in Equiformer paper.

You can check the codebase of Equiformer for both use cases: https://github.com/atomicarchitects/equiformer