Contents Menu Expand Light mode Dark mode Auto light/dark mode
Protein Workshop 0.2.5
Logo
Protein Workshop 0.2.5

Contents:

  • Installation
  • Tutorials
  • Architectural Overview
  • Quickstart
    • Pre-Training
    • Downstream
    • Downloading Datasets

Default Configs

  • Config Templates
  • Dataset
  • Models
  • Tasks
  • Features
  • Transforms
  • Metrics
  • Framework Components
    • Callbacks
    • Debug
    • Environment
    • Loggers
    • Hydra Configuration
    • Extras
  • ML Components
    • Trainer
    • Optimiser
    • Schedulers

API Reference

  • protein_workshop.train
  • protein_workshop.datasets
  • protein_workshop.models
  • protein_workshop.tasks
  • protein_workshop.features
  • protein_workshop.utils
  • protein_workshop.constants
  • protein_workshop.types
  • protein_workshop.metrics
Back to top

ML Components#

  • Trainer
    • Default (default)
    • GPU (gpu)
    • CPU (cpu)
    • DDP (ddp)
    • DDP Sim (ddp_sim)
    • MPS (mps)
  • Optimiser
    • ADAM (adam)
    • ADAM-W (adamw)
    • Lion (lion)
  • Schedulers
    • ReduceLROnPlateau (plateau)
    • LinearWarmupCosineDecay (linear_warmup_cosine_decay)
Next
Trainer
Previous
Extras
Copyright © 2023, Arian R. Jamasb
Made with Sphinx and @pradyunsg's Furo