Module | Submodule | in_channel | out_channel | kernal_size | stride | padding |
---|---|---|---|---|---|---|
Encoder | Downscale | 1 | 8 | 2 | 2 | 0 |
ConvGRU | 8 | 64 | 3 | 1 | 1 | |
Downscale | 64 | 128 | 2 | 2 | 0 | |
ConvGRU | 128 | 128 | 3 | 1 | 1 | |
Downscale | 128 | 128 | 3 | 2 | 1 | |
ConvGRU | 128 | 128 | 3 | 1 | 1 | |
Forecaster | ConvGRU | 128 | 128 | 3 | 1 | 1 |
Upscale | 128 | 128 | 3 | 2 | 1 | |
ConvGRU | 128 | 128 | 3 | 1 | 1 | |
Upscale | 128 | 64 | 2 | 2 | 0 | |
ConvGRU | 64 | 64 | 3 | 1 | 1 | |
Upscale | 64 | 8 | 2 | 2 | 0 | |
Predict-conv1 | 8 | 8 | 3 | 1 | 1 | |
Predict-conv2 | 8 | 1 | 1 | 1 | 0 |