longthangvu commited on
Commit
cda8791
·
1 Parent(s): fd64a47

Add LinearPFN inference weights

Browse files
Files changed (3) hide show
  1. README.md +22 -0
  2. best_model.pt +3 -0
  3. run_config.json +48 -0
README.md ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: pytorch
3
+ tags:
4
+ - time-series
5
+ - forecasting
6
+ - linearpfn
7
+ ---
8
+
9
+ # LinearPFN
10
+
11
+ Inference weights for the from_paper LinearPFN model.
12
+
13
+ - seq_len: 96
14
+ - pred_len: 96
15
+ - checkpoint: best_model.pt
16
+ - config: run_config.json
17
+
18
+ Use from the LinearPFN repo:
19
+
20
+ ```bash
21
+ bash scripts/test_tslib.sh --load_from hf
22
+ ```
best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e5b8146ca763805314b853123e0adc307e97c3eba2fb56daf308d1e1e1e43e88
3
+ size 127130301
run_config.json ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "args": {
3
+ "model": "LinearPFN",
4
+ "seq_len": 96,
5
+ "pred_len": 96,
6
+ "d_model": 512,
7
+ "d_ff": 2048,
8
+ "L_blk": 10,
9
+ "n_heads": 8,
10
+ "dropout": 0.1,
11
+ "data_version": "3",
12
+ "c_min": 32,
13
+ "c_max": 1536
14
+ },
15
+ "notes": "from_paper",
16
+ "model_params": {
17
+ "L": 96,
18
+ "H": 96,
19
+ "dropout": 0.1,
20
+ "d": 512,
21
+ "d_ff": 2048,
22
+ "L_blk": 10,
23
+ "n_heads": 8
24
+ },
25
+ "data": {
26
+ "shards_dir": "./series_bank/from_paper",
27
+ "L": 96,
28
+ "H": 96,
29
+ "C_range": [
30
+ 32,
31
+ 1536
32
+ ],
33
+ "Q_range": [
34
+ 1,
35
+ 32
36
+ ]
37
+ },
38
+ "train_config": {
39
+ "total_tasks": 500000,
40
+ "tasks_per_epoch": 2500,
41
+ "validate_every": 4,
42
+ "save_every": 40,
43
+ "log_every": 250,
44
+ "n_val_tasks": 500,
45
+ "early_stopping_patience": 10,
46
+ "warmup_tasks": 50000
47
+ }
48
+ }