ceyron commited on
Commit
2669d48
·
1 Parent(s): 96b0b1f

Include all postprocessing scripts

Browse files
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