MatterTune

Getting Started

  • Introduction
  • Motivation
  • Installation Guide

User Guide

  • Datasets
  • Fine-Tuning a Pre-trained Model
  • Model Usage Guide
  • Training Configuration Guide
  • Normalization
  • Advanced: Lightning Integration

Advanced Features

  • Accelerate Foundation Models via Message-Passing Pruning and Graph Partitioning

Model Backbones

  • JMP Backbone
  • M3GNet Backbone
  • ORB Backbone
  • EquiformerV2 Backbone
  • MatterSim Backbone

Development

  • API Reference
    • mattertune.backbones
    • mattertune.callbacks
      • mattertune.callbacks.early_stopping
      • mattertune.callbacks.ema
      • mattertune.callbacks.model_checkpoint
      • mattertune.callbacks.multi_gpu_writer
    • mattertune.data
    • mattertune.finetune
    • mattertune.loggers
    • mattertune.main
    • mattertune.normalization
    • mattertune.registry
    • mattertune.wrappers
  • Contributing to MatterTune
  • License Information
MatterTune
  • API Reference
  • mattertune.callbacks
  • View page source

mattertune.callbacks

Modules

early_stopping

ema

model_checkpoint

multi_gpu_writer

Previous Next

© Copyright 2024, MatterTune Team.

Built with Sphinx using a theme provided by Read the Docs.