Skip to content

Model Zooยค

Here we document the performance of various models currently implemented with PDEArena. The tables below provide results and useful statistics about training and inference.

Model Name Num. Params Model Size (MB) Peak GPU Memory (MB) Forward Time (s) Forward+Backward Time (s)
DilResNet128 DilResNet-128 4.2 M 16.7 4849.01 0.118 0.342
DilResNet-128-norm DilResNet-128-norm 4.2 M 16.7 6921.0 0.183 0.423
FNO128-8modes16 FNO-128-16m 134 M 537.5 2950.0 0.059 0.172
FNO128-8modes8 FNO-128-8m 33.7 M 134.9 2160.5 0.056 0.161
FNO128-4modes16 FNOs-128-16m 67.2 M 268.8 1850.95 0.031 0.089
FNO128-4modes32 FNOs-128-32m 268 M 1074.1 3416.53 0.036 0.118
FNO64-4modes32 FNOs-64-32m 67.1 M 268.5 1204.57 0.016 0.050
FNO96-4modes32 FNOs-96-32m 151 M 604.2 2179.0 0.026 0.080
ResNet128 ResNet-128 1.2 M 4.9 2491.01 0.043 0.093
ResNet256 ResNet-256 4.9 M 19.4 4941.51 0.118 0.261
U-F1Netmodes16 U-FNet1-16m 160 M 643.6 4034.6 0.082 0.196
U-F1Netmodes16-1x1 U-FNet1-16m-1x1 160 M 643.6 4037.58 0.081 0.195
U-F1Netmodes8 U-FNet1-8m 148 M 593.3 3934.09 0.081 0.195
U-F1Netmodes8-1x1 U-FNet1-8m-1x1 148 M 593.3 3936.99 0.081 0.194
U-F2Netmodes16,8 U-FNet2-16m 175 M 700.5 4146.64 0.083 0.200
U-F2Netmodes16,8-1x1 U-FNet2-16m-1x1 175 M 700.5 4150.08 0.083 0.199
U-F2Netmodes16,16 U-FNet2-16mc 219 M 876.7 4493.39 0.084 0.204
U-F2Netmodes8,4 U-FNet2-8m 151 M 606.2 3961.9 0.082 0.198
U-F2Netmodes8,4-1x1 U-FNet2-8m-1x1 151 M 606.1 3961.2 0.082 0.197
U-F2Netmodes8,8 U-FNet2-8mc 162 M 650.2 4046.27 0.082 0.199
U-F2Netatt,modes16,8 U-FNet2attn-16m 179 M 717.3 4228.26 0.085 0.206
U-F2Netatt,modes16,8-1x1 U-FNet2attn-16m-1x1 179 M 717.3 4224.25 0.085 0.205
U-F3Netmodes16,8,4 U-FNet3-16m 187 M 751.9 4246.78 0.083 0.203
U-F3Netmodes16,8,4-1x1 U-FNet3-16m-1x1 187 M 751.9 4249.58 0.083 0.201
U-F3Netmodes8,4,2 U-FNet3-8m 164 M 657.5 4059.5 0.083 0.201
U-F3Netmodes8,4,2-1x1 U-FNet3-8m-1x1 164 M 657.5 4060.67 0.083 0.200
UNO128 UNO-128 440 M 1761.8 5511.0 0.158 0.341
UNO64 UNO-64 110 M 440.5 1931.67 0.065 0.134
U-Net2015128 Unet2015-128 124 M 496.5 3001.0 0.042 0.117
U-Net201564 Unet2015-64 31.0 M 124.2 1303.97 0.013 0.037
U-Netbase128 Unetbase-128 124 M 496.6 2999.0 0.056 0.134
U-Netbase64 Unetbase-64 31.1 M 124.2 1276.35 0.021 0.046
U-Netmod64 Unetmod-64 144 M 577.1 3904.68 0.079 0.186
U-Netmod64-1x1 Unetmod-64-1x1 144 M 577.0 3907.24 0.079 0.185
U-Netatt64 Unetmodattn-64 148 M 593.9 3980.78 0.081 0.192
U-Netatt64-1x1 Unetmodattn-64-1x1 148 M 593.8 3976.78 0.081 0.191
  • Date Created: 2022-11-12 18:46:37.780512
  • GPU: Tesla V100-PCIE-16GB