mattertune.registry

Module Attributes

backbone_registry	Registry for backbone modules.
data_registry	Registry for data modules.

mattertune.registry.backbone_registry = Registry(base_cls=<class 'mattertune.finetune.base.FinetuneModuleBaseConfig'>, discriminator='name', config={'duplicate_tag_policy': 'error'}, _elements=[_RegistryEntry(tag='eqV2', cls=<class 'mattertune.backbones.eqV2.model.EqV2BackboneConfig'>), _RegistryEntry(tag='jmp', cls=<class 'mattertune.backbones.jmp.model.JMPBackboneConfig'>), _RegistryEntry(tag='m3gnet', cls=<class 'mattertune.backbones.m3gnet.model.M3GNetBackboneConfig'>), _RegistryEntry(tag='orb', cls=<class 'mattertune.backbones.orb.model.ORBBackboneConfig'>)], _on_register_callbacks=[<function Registry.rebuild_on_registers.<locals>._rebuild>])

Registry for backbone modules.

mattertune.registry.data_registry = Registry(base_cls=<class 'nshconfig._src.config.Config'>, discriminator='type', config={'duplicate_tag_policy': 'error'}, _elements=[_RegistryEntry(tag='json', cls=<class 'mattertune.data.json_data.JSONDatasetConfig'>), _RegistryEntry(tag='matbench', cls=<class 'mattertune.data.matbench.MatbenchDatasetConfig'>), _RegistryEntry(tag='mp', cls=<class 'mattertune.data.mp.MPDatasetConfig'>), _RegistryEntry(tag='omat24', cls=<class 'mattertune.data.omat24.OMAT24DatasetConfig'>), _RegistryEntry(tag='xyz', cls=<class 'mattertune.data.xyz.XYZDatasetConfig'>), _RegistryEntry(tag='db', cls=<class 'mattertune.data.db.DBDatasetConfig'>), _RegistryEntry(tag='mptraj', cls=<class 'mattertune.data.mptraj.MPTrajDatasetConfig'>)], _on_register_callbacks=[<function Registry.rebuild_on_registers.<locals>._rebuild>, <function Registry.rebuild_on_registers.<locals>._rebuild>, <function Registry.rebuild_on_registers.<locals>._rebuild>])

Registry for data modules.