目录

├── data
│   ├── htmp_data
│   ├── train_data
│   │   ├── ACCESS-CM2
│   │   ├── CCSM4
│   │   ├── CESM1-CAM5
│   │   ├── ...
│   │   └── obs
│   └── var_data

python train.py

python train.py \
    --save_ckpt true \
    --load_ckpt false \
    --save_ckpt_path ./checkpoints \
    --load_ckpt_path ./checkpoints/exp2_aftertrain/enso_float16.ckpt \
    --save_data true\
    --load_data_path ./data \
    --save_data_path ./data \
    --save_figure true \
    --figures_path ./figures \
    --log_path ./logs \
    --print_interval 10 \
    --lr 0.01 \
    --epochs 20 \
    --batch_size 400 \
    --skip_aftertrain false \
    --epochs_after 5 \
    --batch_size_after 30 \
    --lr_after 1e-6 \
    --download_data enso \
    --force_download false \
    --amp_level O3 \
    --device_id 0 \
    --mode 0

├── enso
│   ├── checkpoints                       # checkpoint文件
│   ├── data                              # 数据文件
│   │   ├── htmp_data                     # 验证结果的保存路径
│   │   ├── var_data                      # 验证数据集
│   │   └── train_data                    # 训练数据集
│   ├── figures                           # 结果图片
│   ├── logs                              # 日志文件
│   ├── src                               # 源代码
│   │   ├── network.py                    # 网络架构
│   │   ├── plot.py                       # 绘制结果
│   │   └── process.py                    # 数据处理
│   ├── config.yaml                       # 超参数配置
│   ├── README.md                         # 英文模型说明
│   ├── README_CN.md                      # 中文模型说明
│   ├── train.py                          # python训练脚本
│   └── eval.py                           # python评估脚本

# python train.py
...
epoch: 1 step: 1, loss is 0.9130635857582092
epoch: 1 step: 2, loss is 1.0354164838790894
epoch: 1 step: 3, loss is 0.8914494514465332
epoch: 1 step: 4, loss is 0.9377754330635071
epoch: 1 step: 5, loss is 1.0472232103347778
epoch: 1 step: 6, loss is 1.0421113967895508
epoch: 1 step: 7, loss is 1.100639820098877
epoch: 1 step: 8, loss is 0.9849204421043396
...

python eval.py