Upload PatchTSMixerForForecasting

config.json

@@ -4,7 +4,7 @@
     "PatchTSMixerForForecasting"
   ],
   "channel_consistent_masking": true,
-  "d_size": "4D",
+  "distribution_output": "student_t",
   "dropout": 0.5,
   "expansion_factor": 3,
   "forecast_channel_indices": null,
@@ -12,9 +12,10 @@
   "gated_attn": true,
   "head_agg": "max_pool",
   "head_dropout": 0.7,
-  "in_channels": 7,
   "init_std": 0.02,
+  "input_size": 7,
   "learn_pe": false,
+  "loss": "mse",
   "mask_patch_ratios": [
     1,
     1
@@ -34,6 +35,7 @@
   "norm_mlp": "LayerNorm",
   "num_features": 48,
   "num_layers": 2,
+  "num_parallel_samples": 100,
   "num_patches": 32,
   "output_range": null,
   "patch_last": true,
@@ -41,12 +43,13 @@
   "pe": "zeros",
   "post_init": false,
   "revin": true,
-  "seed_number": 42,
+  "scaling": "std",
+  "seed_number": null,
   "self_attn": false,
   "self_attn_heads": 1,
   "seq_len": 512,
   "stride": 16,
   "torch_dtype": "float32",
-  "transformers_version": "4.33.0.dev0",
+  "transformers_version": "4.35.0.dev0",
   "use_pe": false
 }

generation_config.json

@@ -0,0 +1,4 @@
+{
+  "_from_model_config": true,
+  "transformers_version": "4.35.0.dev0"
+}

pytorch_model.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:552e937b569f48b4e2bac9f956be93b4e2ca62025f2e9d9c5320985763cbf152
-size 796665
+oid sha256:684fb8f2ee6e38ab430e77a4e8cd86028c34a150d1251f7d2a624f4de0e8c9c8
+size 796601