Skip to content

Samudra

m2lines/Samudra

Home
Getting Started
Getting Started
- Installation
- Quick Start
Concepts
Concepts
- Architecture
- Data
- How to Run
- Torch
Development
Development
- Contributing
API Reference
API Reference
- Models
  Models
- Datasets
- Training
  Training
  - Stepper
  - Train
  - Eval
- Configuration
- Utilities
  Utilities
  - Data Utils
  - Distributed
  - Loss
  - EMA
  - Schedule Schedule
    On this page
    
    schedule
    
    CosineSchedulerConfig
    
    CosineWithTailSchedulerConfig
    
    CosineWithWarmupConfig
  - Samplers
  - Writer
- Visualization

Schedule¶

`samudra.utils.schedule` ¶

`CosineSchedulerConfig` ¶

Bases: BaseModel

Cosine scheduler; see pytorch CosineAnnealingLR.

`CosineWithTailSchedulerConfig` ¶

Bases: BaseModel

Cosine scheduler which goes to tail_lr for the last tail_epochs.

`CosineWithWarmupConfig` ¶

Bases: BaseModel

Cosine scheduler which goes from warmup_lr to the default lr for the first warmup_epochs.