ajati commited on
Commit
9906e60
1 Parent(s): cca8abd

Upload PatchTSMixerForForecasting

Browse files
Files changed (2) hide show
  1. config.json +52 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "ibm/patchtsmixer-etth1-pretrain",
3
+ "architectures": [
4
+ "PatchTSMixerForForecasting"
5
+ ],
6
+ "channel_consistent_masking": true,
7
+ "d_size": "4D",
8
+ "dropout": 0.5,
9
+ "expansion_factor": 3,
10
+ "forecast_channel_indices": null,
11
+ "forecast_len": 96,
12
+ "gated_attn": true,
13
+ "head_agg": "max_pool",
14
+ "head_dropout": 0.7,
15
+ "in_channels": 7,
16
+ "init_std": 0.02,
17
+ "learn_pe": false,
18
+ "mask_patch_ratios": [
19
+ 1,
20
+ 1
21
+ ],
22
+ "mask_patches": [
23
+ 2,
24
+ 3
25
+ ],
26
+ "mask_ratio": 0.5,
27
+ "mask_type": "random",
28
+ "mask_value": 0,
29
+ "masked_loss": true,
30
+ "mode": "common_channel",
31
+ "model_type": "patchtsmixer",
32
+ "n_classes": 3,
33
+ "n_targets": 3,
34
+ "norm_mlp": "LayerNorm",
35
+ "num_features": 48,
36
+ "num_layers": 2,
37
+ "num_patches": 32,
38
+ "output_range": null,
39
+ "patch_last": true,
40
+ "patch_len": 16,
41
+ "pe": "zeros",
42
+ "post_init": false,
43
+ "revin": true,
44
+ "seed_number": 42,
45
+ "self_attn": false,
46
+ "self_attn_heads": 1,
47
+ "seq_len": 512,
48
+ "stride": 16,
49
+ "torch_dtype": "float32",
50
+ "transformers_version": "4.33.0.dev0",
51
+ "use_pe": false
52
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:552e937b569f48b4e2bac9f956be93b4e2ca62025f2e9d9c5320985763cbf152
3
+ size 796665