CSE 803 Final Project

A space for code and data associated with the final project in CSE 803 with Professor Xiaoming Liu @ MSU, Fall 2021.

Installation

Using conda for the larger packages

If you're using the HPCC be sure to uncomment the cudatoolkit line in the environment.yml. If you're on a different machine with CUDA change the toolkit version to match your system.

conda env create -f environment.yml
conda activate CVProj

If you have to make changes to the environment you can update the configuration with

conda env update -f environment.yml

Data

Download the era5.nc data file here, and save it into the root directory of this repository to ensure all code is able to find the data. This file contains monthly average surface temperature (t2m) and precipitation (tp) on a 0.1 degree grid over the United States from 1950-2020, and was extracted from the ERA5 Reanalysis Dataset.

Testing

To test a checkpointed model use

python test.py --model <MODEL NAME> --checkpoint </path/to/checkpoint.ckpt> --gpus=1

Group Members

• Andrew McDonald
• Drew Hayward
• Tyler Lovell
• Sanjeev Thenkarai Lakshmi Narasimhan