Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
* Scripts for table 1 * Fixed typos in table1.py * Temp fix * iter module working with new refactor * Checkpoint Tried to get ShardDataset working. Switching to IterDataset. * Update table 1 code * Adding table experiments for other architectures * Update ViT architecture * Add notebook * Checkpoint * Checkpointing * Updated log dir for checkpoints * Checkpoint * Fix mse, rmse, and acc metrics * Update acc code * Reduce precision, change early stopping metric * Fixes * add lr monitor, modify checkpointing, use ddp instead of ddp_spawn * modify logging keys to separate train/val/test * use normalized lat_mse as a validation metric * change training setting * Checkpoint * Downscaling * DL models for downscaling * Pearson now does not rely on scipy * ViT experiment and PRISM support * Multiple transformation bug fix * fix duplicate denormalization bug * add ReduceLROnPlateau scheduler * update gitignore * fix resnet to match with weatherbench implementation * current best setting * update projection and prediction layers for unet * update run scripts for resnet and unet * avoid spiking loss * update vit models and scripts * Downscaling exp changes * Downscaling changes * rollback resnet and unet architectures * Merge Tung's changes * Downscaling updates * use 16 precision * CMIP6 running with iterdatamodule * add continuous data modules * update nc2npz to accumulate precipitation data * merge tung exps (#105) * avoid spiking loss * update vit models and scripts * rollback resnet and unet architectures * use 16 precision * add continuous data modules * update nc2npz to accumulate precipitation data --------- Co-authored-by: Tung Nguyen <[email protected]> * ERA5 downscaling updates * use 1 worker, somehone increasing # workers affects performance * update scripts to be consistent * prepare iterative exps * Downscaling checkpoint: PRISM implemented * evaluate iterative models * summarize results * update json results and add plot scripts * Add probabilistic metrics * Add class metrics * Add rank histogram visualization * update plot script * add climatebench experiments * Disable all data tests * Formatting * Fix trainer * Cleanup files * Moved processing scripts, added PRISM processing * Cleanup data downloading * Add CLI utility for downloads, processing scripts * Combine iter modules * Add data modules * Reorganize files * Remove unused notebooks * Cleaning utilities * Add visualizations for bias and mean bias * Add rank histogram visualization * Organize experiments, add ERA5 crop script * Clean up downscaling and climate projection * Add forecasting baselines * Cleanup forecasting experiments * Removed unneeded tables * Add climate projection experiments * Iterative forecasting experiments * Iterative forecasting experiments * Add extreme ERA5 * Changes for quickstart notebook * Update notebooks * Formatting * Update era5_constants.py Add units for variables * Update cmip6_constants.py Add units for variables. * Add units for visualizations * Updates for notebook * Update docs and notebook * Update docs * Finish tasks and datasets * Finish docs * Update configs * Fix pyproject.toml * Remove rasterio version for rtd * Change rtd python version * Add OS * Add images * v1 release --------- Co-authored-by: Prakhar Sharma <[email protected]> Co-authored-by: tung-nd <[email protected]> Co-authored-by: Prakhar Sharma <[email protected]>
- Loading branch information