Skip to content

Ocean Emulators

Open-Athena/Ocean_Emulator

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
  - Base Model
  - Samudra
  - FOMO
  - Modules
- 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¶

`ocean_emulators.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.