Upload checkpoint

architecture-MIFNO3D-moment-dv16-16layers-S32-T320-learningrate0p0004-Ntrain27000-batchsize16-normedsource-normedtraces-epochs300.json

@@ -0,0 +1 @@
+{"architecture": {"model type": "MIFNO", "input dim": 4, "output dim": 1, "source dim": 9, "nlayers": 16, "branching index": 4, "dv": 16, "list D1": [32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32], "list D2": [32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32], "list D3": [32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 32, 64, 128, 256, 320], "list M1": [16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16], "list M2": [16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16], "list M3": [16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 16, 32, 32, 32], "padding": 0}, "optimization": {"weight decay": 1e-05, "patience": 60, "epochs": 300, "learning rate": 0.0004, "loss weights (L1, L2)": [1.0, 0.0]}, "data": {"Ntrain": 27000, "Nval": 3000, "batch size": 16, "source orientation": "moment", "normalize source": true, "transform position": [9600.0, 9600.0, -9600.0], "normalize traces": "distance_Vs", "data train": ["../data/formatted/HEMEWS3D_S32_Z32_T320_fmax5_rot0_train"], "data val": ["../data/formatted/HEMEWS3D_S32_Z32_T320_fmax5_rot0_val"]}}

bestmodel-MIFNO3D-moment-dv16-16layers-S32-T320-learningrate0p0004-Ntrain27000-batchsize16-normedsource-normedtraces-epochs300.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9ef0fb16cd0512241c3c69849b63984badcc7f1f7ae76f155b6be8629bf6c039
+size 13676955