Test turning all grad checkpointings off#1168

Closed

javak87 wants to merge 22 commits intoecmwf:developfrom

javak87:javad/dev/cond_checkpoint_all-1141

Commits on Oct 24, 2025

add embed_gradient_checkpoint_mode to config
Javad Kasravi
committed
add embed_gradient_checkpoint_mode condition to forward_channels
Javad Kasravi
committed
add embed_gradient_checkpoint_mode to args
Javad Kasravi
committed
test pipeline for embed_gradient_checkpoint_mode true
Javad Kasravi
committed
test pipeline for embed_gradient_checkpoint_mode false
Javad Kasravi
committed

Commits on Oct 25, 2025

add ae_local_blocks_grdient_checkpoint_mode to config
Javad Kasravi
committed
add ae_local_blocks_grdient_checkpoint_mode cond
Javad Kasravi
committed
add ae_adapter_grdient_checkpoint_mode
Javad Kasravi
committed
add assimilate_global_gradient_checkpoint_mode to config
Javad Kasravi
committed
add pred_gradient_checkpoint_mode to config
Javad Kasravi
committed

Commits on Oct 27, 2025

Merge branch 'develop' of https://github.com/javak87/WeatherGenerator into javad/dev/cond_checkpoint_embed_transformer
Javad Kasravi
committed
ruff embeddings.py
Javad Kasravi
committed
remove ae_local_blocks_grdient_checkpoint_mode for __init__
Javad Kasravi
committed
ruff the code
Javad Kasravi
committed
ruff the code
Javad Kasravi
committed
ruff the code
Javad Kasravi
committed
ruff the code
Javad Kasravi
committed
Merge branch 'javad/dev/cond_checkpoint_ae_local-1141' into javad/dev/cond_checkpoint_all-1141
Javad Kasravi
committed
Merge branch 'javad/dev/cond_checkpoint_assimilate_global-1141' into javad/dev/cond_checkpoint_all-1141
Javad Kasravi
committed
Merge branch 'javad/dev/cond_checkpoint_embed_transformer-1141' into javad/dev/cond_checkpoint_all-1141
Javad Kasravi
committed
Merge branch 'javad/dev/cond_checkpoint_predict-1141' into javad/dev/cond_checkpoint_all-1141
Javad Kasravi
committed

Commits on Oct 28, 2025

Merge branch 'develop' into javad/dev/cond_checkpoint_all-1141
javak87
authored