xnn.cnn.models#

Modules

base

Shared body of the grid-input models of the cnn family.

cnn3d

Conventional 3D CNN over voxelized atomic environments.

spherical

Spherical CNN (Cohen, Geiger, Koehler, Welling, ICLR 2018).

steerable

3D steerable CNN (Weiler, Geiger, Welling, Boomsma, Cohen, NeurIPS 2018).