atomgen.models package#
Models module for the AtomGen library.
This module contains the model classes and functions for training and inference.
Submodules#
- atomgen.models.schnet module
SchNetConfigSchNetConfig.vocab_sizeSchNetConfig.hidden_channelsSchNetConfig.num_filtersSchNetConfig.num_interactionsSchNetConfig.num_gaussiansSchNetConfig.cutoffSchNetConfig.interaction_graphSchNetConfig.max_num_neighborsSchNetConfig.readoutSchNetConfig.dipoleSchNetConfig.meanSchNetConfig.stdSchNetConfig.atomrefSchNetConfig.mask_token_idSchNetConfig.pad_token_idSchNetConfig.bos_token_idSchNetConfig.eos_token_id
SchNetModelSchNetPreTrainedModel
- atomgen.models.tokengt module
- atomgen.models.unimolplus module
GaussianLayerInitialStructure2RelaxedEnergyInitialStructure2RelaxedStructureInitialStructure2RelaxedStructureAndEnergyParallelBlockStructure2EnergyStructure2EnergyAndForcesStructure2ForcesTransformerConfigTransformerEncoderTransformerForCoordinateAMTransformerForMaskedAMTransformerModelTransformerPreTrainedModel