difference in CNN models

[Suvi-dha]

Hi,

May I know what changes you have made in different CNN models in your code and how are they different from each other. Which one did you actually use in your implementation? CNN, CNN_B, CNN_B_alpha, CNN_B_scalar.

Also, I want to know, is it possible to modify this code to include one more energy term?

Thanks.

[dmarcosg]

Hello! Sorry again for the confusion. CNN_B refers to the architecture used in the paper, and _alpha and _scalar refer to the ablation studies.

What energy term would you want to add? If it can be written as a linear function of the node positions, it's straight forward to add it.

[Suvi-dha]

I want to add gradient vector flow (external force)