Include all postprocessing scripts
Browse files- postprocessing/adv_rollout_compensate_train_time/img/increasing_resnet_rollout_train_time_compensate.pdf +3 -0
- postprocessing/adv_rollout_compensate_train_time/img/increasing_resnet_rollout_zoomed_compensate.pdf +3 -0
- postprocessing/adv_rollout_compensate_train_time/postprocessing_avd_rollout_compensate_train_time.ipynb +3 -0
- postprocessing/optimization_configuration_ablation/img/optimization_configuration_ablation.pdf +3 -0
- postprocessing/optimization_configuration_ablation/postprocessing_optimization_configuration_ablation.ipynb +3 -0
- postprocessing/parameter_scaling_ablation/get_number_of_params.py +23 -0
- postprocessing/parameter_scaling_ablation/img/parameter_scaling_ablation.pdf +3 -0
- postprocessing/parameter_scaling_ablation/number_of_params.json +1 -0
- postprocessing/parameter_scaling_ablation/postprocessing_parameter_scaling_ablation.ipynb +3 -0
- postprocessing/train_dataset_size_ablation/img/train_dataset_size_ablation.pdf +3 -0
- postprocessing/train_dataset_size_ablation/postprocessing_train_dataset_size_ablation.ipynb +3 -0
postprocessing/adv_rollout_compensate_train_time/img/increasing_resnet_rollout_train_time_compensate.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a29a920fb9057bfc524b50ce4fd2bf038f05652ab2aae544b627c66fe7cc3e9c
|
| 3 |
+
size 15552
|
postprocessing/adv_rollout_compensate_train_time/img/increasing_resnet_rollout_zoomed_compensate.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:90b47cf9e6a3d0562388b1204cceef10ff22bdf8209905cf29dd88cc0bbecb73
|
| 3 |
+
size 16268
|
postprocessing/adv_rollout_compensate_train_time/postprocessing_avd_rollout_compensate_train_time.ipynb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:edc21d2589023c3b2cc3a550113e810bf2a6fb416b70b3d32cb09deb5290b702
|
| 3 |
+
size 95759
|
postprocessing/optimization_configuration_ablation/img/optimization_configuration_ablation.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d3ebf711d15e57c9f940ce909ea02f0970a02c892ee50ba2d6c60310ff6a655b
|
| 3 |
+
size 40259
|
postprocessing/optimization_configuration_ablation/postprocessing_optimization_configuration_ablation.ipynb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:c5bab4a30396a93570b35307b361709985036e2ebf3d39cdb3286cec0656fe84
|
| 3 |
+
size 307371
|
postprocessing/parameter_scaling_ablation/get_number_of_params.py
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
import json
|
| 2 |
+
|
| 3 |
+
import apebench
|
| 4 |
+
|
| 5 |
+
# All scenarios are single channel
|
| 6 |
+
scene = apebench.scenarios.difficulty.Advection()
|
| 7 |
+
|
| 8 |
+
param_dict = {}
|
| 9 |
+
|
| 10 |
+
for net in [
|
| 11 |
+
*[f"Conv;{2**hc_exp:03d};10;relu" for hc_exp in range(9 + 1)],
|
| 12 |
+
*[f"UNet;{2**hc_exp:03d};2;relu" for hc_exp in range(6 + 1)],
|
| 13 |
+
*[f"Res;{2**hc_exp:03d};8;relu" for hc_exp in range(9 + 1)],
|
| 14 |
+
*[f"FNO;12;{2**hc_exp:03d};4;relu" for hc_exp in range(5 + 1)],
|
| 15 |
+
*[f"Dil;2;{2**hc_exp:03d};2;relu" for hc_exp in range(8 + 1)],
|
| 16 |
+
]:
|
| 17 |
+
num_params = scene.get_parameter_count(net)
|
| 18 |
+
|
| 19 |
+
param_dict[net] = num_params
|
| 20 |
+
|
| 21 |
+
|
| 22 |
+
with open("number_of_params.json", "w") as f:
|
| 23 |
+
json.dump(param_dict, f)
|
postprocessing/parameter_scaling_ablation/img/parameter_scaling_ablation.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8b0d64853b3e02823a5beeeb675554b89fe8bdc74671a199deb1cdfa426b0fa8
|
| 3 |
+
size 22804
|
postprocessing/parameter_scaling_ablation/number_of_params.json
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
{"Conv;001;10;relu": 44, "Conv;002;10;relu": 141, "Conv;004;10;relu": 497, "Conv;008;10;relu": 1857, "Conv;016;10;relu": 7169, "Conv;032;10;relu": 28161, "Conv;064;10;relu": 111617, "Conv;128;10;relu": 444417, "Conv;256;10;relu": 1773569, "Conv;512;10;relu": 7086081, "UNet;001;2;relu": 254, "UNet;002;2;relu": 873, "UNet;004;2;relu": 3209, "UNet;008;2;relu": 12273, "UNet;016;2;relu": 47969, "UNet;032;2;relu": 189633, "UNet;064;2;relu": 754049, "Res;001;8;relu": 68, "Res;002;8;relu": 231, "Res;004;8;relu": 845, "Res;008;8;relu": 3225, "Res;016;8;relu": 12593, "Res;032;8;relu": 49761, "Res;064;8;relu": 197825, "Res;128;8;relu": 788865, "Res;256;8;relu": 3150593, "Res;512;8;relu": 12592641, "FNO;12;001;4;relu": 108, "FNO;12;002;4;relu": 415, "FNO;12;004;4;relu": 1629, "FNO;12;008;4;relu": 6457, "FNO;12;016;4;relu": 25713, "FNO;12;032;4;relu": 102625, "Dil;2;001;2;relu": 64, "Dil;2;002;2;relu": 187, "Dil;2;004;2;relu": 613, "Dil;2;008;2;relu": 2185, "Dil;2;016;2;relu": 8209, "Dil;2;032;2;relu": 31777, "Dil;2;064;2;relu": 124993, "Dil;2;128;2;relu": 495745, "Dil;2;256;2;relu": 1974529}
|
postprocessing/parameter_scaling_ablation/postprocessing_parameter_scaling_ablation.ipynb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:edb3b0202bfb578659d01493de7ede156bbc62d500c10a3c8a708d69658be797
|
| 3 |
+
size 104646
|
postprocessing/train_dataset_size_ablation/img/train_dataset_size_ablation.pdf
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:f39b4c761b4572e9afff5a61d2f30bb3a56a2f3521af429504c5b1a291911080
|
| 3 |
+
size 48761
|
postprocessing/train_dataset_size_ablation/postprocessing_train_dataset_size_ablation.ipynb
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:bcab10cc5a629aeb67e48ca647eb46f083708c78cb5a0e89f51d61f0a944d30d
|
| 3 |
+
size 262772
|