512-96-r2

config.json

@@ -0,0 +1,59 @@
+{
+  "adaptive_patching_levels": 3,
+  "architectures": [
+    "TinyTimeMixerForPrediction"
+  ],
+  "categorical_vocab_size_list": null,
+  "context_length": 512,
+  "d_model": 192,
+  "d_model_scale": 3,
+  "decoder_adaptive_patching_levels": 0,
+  "decoder_d_model": 128,
+  "decoder_d_model_scale": 2,
+  "decoder_mode": "common_channel",
+  "decoder_num_layers": 2,
+  "decoder_raw_residual": false,
+  "distribution_output": "student_t",
+  "dropout": 0.4,
+  "enable_forecast_channel_mixing": false,
+  "exogenous_channel_indices": null,
+  "expansion_factor": 2,
+  "fcm_context_length": 1,
+  "fcm_gated_attn": true,
+  "fcm_mix_layers": 3,
+  "fcm_prepend_past": true,
+  "fcm_use_mixer": true,
+  "frequency_token_vocab_size": 5,
+  "gated_attn": true,
+  "head_dropout": 0.4,
+  "init_embed": "pytorch",
+  "init_linear": "pytorch",
+  "init_processing": true,
+  "init_std": 0.02,
+  "loss": "mse",
+  "mode": "common_channel",
+  "model_type": "tinytimemixer",
+  "norm_eps": 1e-05,
+  "norm_mlp": "LayerNorm",
+  "num_input_channels": 1,
+  "num_layers": 2,
+  "num_parallel_samples": 100,
+  "num_patches": 8,
+  "patch_last": true,
+  "patch_length": 64,
+  "patch_stride": 64,
+  "positional_encoding_type": "sincos",
+  "post_init": false,
+  "prediction_channel_indices": null,
+  "prediction_filter_length": null,
+  "prediction_length": 96,
+  "resolution_prefix_tuning": false,
+  "scaling": "std",
+  "self_attn": false,
+  "self_attn_heads": 1,
+  "stride_ratio": 1,
+  "torch_dtype": "float32",
+  "transformers_version": "4.37.2",
+  "use_decoder": true,
+  "use_positional_encoding": false
+}

generation_config.json

@@ -0,0 +1,4 @@
+{
+  "_from_model_config": true,
+  "transformers_version": "4.37.2"
+}

model.safetensors

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

training_args.bin

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