allegro icon indicating copy to clipboard operation
allegro copied to clipboard

Question about the Tensor Product module.

Open floatingCatty opened this issue 1 year ago • 0 comments

Hello, I have been reading the code of allegro and trying to implement an E3-equivariant model recently. I am confused about the tensor product construction process in the allegro model. For example, in nn/_strided/_linear, and nn/_stride/_contract, some modules seem to realize the module to build tensor-production and linear module in e3nn. Still, when I compare the code with the e3nn module, I find many differences, such as the normalization, and some parameters such as the sparse mode.

So my question is, what are the functional differences comparing the tensor production module in allegro/nn concerning e3nn?

floatingCatty avatar Dec 10 '23 07:12 floatingCatty