Upload 6 files

.gitattributes

@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+last.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text
+weights.data-00000-of-00001 filter=lfs diff=lfs merge=lfs -text

checkpoint

@@ -0,0 +1,2 @@
+model_checkpoint_path: "last"
+all_model_checkpoint_paths: "last"

config.json

@@ -0,0 +1,57 @@
+{
+  "encoder_config": {
+    "row_wise_ff_num_layers": 2,
+    "num_heads": 12,
+    "num_layers": 4,
+    "d_model": 384,
+    "m_inducing_points": 64,
+    "k_seed_vectors": 50,
+    "dropout_rate": 0.1
+  },
+  "decoder_config": {
+    "__class__": "decoder_with_regression_config",
+    "d_model": 384,
+    "num_heads": 12,
+    "d_ff": 2048,
+    "num_layers": 4,
+    "max_len": 50,
+    "mode":"concat"
+  },
+  "encoder_expr":{
+    "d_model": 384,
+    "num_heads": 12,
+    "d_ff": 1024,
+    "num_layers": 4,
+    "max_len": 50,
+    "dropout_rate": 0.1
+  },  
+  "loss":"mse",
+  "reg_head_num_layers": 2,
+  "reg_head_dim": 64,
+  "dataset_config": {
+    "num_of_not_seen": 3,
+    "batch_size": 128,
+    "sample_points": false,
+    "test_size": 10000,
+    "dataset_size": 10000000,
+    "variables": ["x", "y"],
+    "num_points": 200,
+    "extended_representation": true
+  },
+  "max_epoch": 300,
+  "input_regularizer": "random_noise_regularizer()",
+  "callback_config": {
+    "greedy_search": {
+      "max_num": 32,
+      "evaluate_each": 0
+    },
+    "beam_search": {
+      "max_num": 32,
+      "evaluate_each": 30,
+      "beam_width": 256
+    },
+    "regression_delay": 100
+  },
+  "regression_lambda": 0.0,
+  "label_smoothing": 0.1
+}

last.data-00000-of-00001

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8fb515aa3fed6d01dc3b0e18e8f38210e800b645a2d8520692b58f6d7f1f2215
+size 1080686209

weights.data-00000-of-00001

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