ACE2-SOM / inference_config.yaml
spencerkclark's picture
Add checkpoint and update README.md (#1)
b96e8d7 verified
raw
history blame contribute delete
614 Bytes
experiment_dir: /output_directory
n_forward_steps: 400
forward_steps_in_memory: 50
checkpoint_path: /ace2_som_ckpt.tar
logging:
log_to_screen: true
log_to_wandb: false
log_to_file: true
project: ace
initial_condition:
path: /initial_conditions/1xCO2/ic.nc
start_indices:
times:
- "2032-01-01T00:00:00"
forcing_loader:
dataset:
data_path: /forcing_data/1xCO2
n_repeats: 1 # Use this to extend the four years of forcing data to the desired length
num_data_workers: 4
data_writer:
save_prediction_files: true
save_monthly_files: false
names: ["TMP2m", "VGRD10m", "PRATEsfc"]