Is this a new feature, an improvement, or a change to existing functionality?
Improvement
How would you describe the priority of this feature request
Medium
Please provide a clear description of problem you would like to solve.
The current training pipeline (likely EMAHook) is not compatible with the case when cuequivariance is enabled with conv_fusion set to True. Enabling conv_fusion is likely very helpful in accelerating model training. In fact, that is the default setting in the official mace training cli.
Describe any alternatives you have considered
No response
Is this a new feature, an improvement, or a change to existing functionality?
Improvement
How would you describe the priority of this feature request
Medium
Please provide a clear description of problem you would like to solve.
The current training pipeline (likely EMAHook) is not compatible with the case when cuequivariance is enabled with conv_fusion set to True. Enabling conv_fusion is likely very helpful in accelerating model training. In fact, that is the default setting in the official mace training cli.
Describe any alternatives you have considered
No response