yushuang88 commited on
Commit
6c0f87b
·
verified ·
1 Parent(s): dbac203

Update model configuration

Browse files
Files changed (1) hide show
  1. config.json +4 -49
config.json CHANGED
@@ -1,51 +1,6 @@
1
  {
2
- "architectures": [
3
- "GraphViT"
4
- ],
5
- "framework": "pytorch",
6
- "model_name": "EagleMeshTransformer",
7
- "model_type": "eaglemeshtransformer",
8
- "source_config": "config/config.yaml",
9
- "model": {
10
- "name": "EagleMeshTransformer",
11
- "architecture": "GraphViT",
12
- "state_size": 4,
13
- "w_size": 32
14
- },
15
- "datapipe": {
16
- "source": {
17
- "data_dir": "./data/fake/Eagle_dataset",
18
- "cluster_dir": "./data/fake/Eagle_dataset",
19
- "splits_dir": "./data/fake/splits"
20
- },
21
- "verbose": false,
22
- "data": {
23
- "window_length_train": 3,
24
- "window_length_val": 3,
25
- "window_length_test": 3,
26
- "n_cluster": 1,
27
- "normalized": true,
28
- "type_as_onehot": true,
29
- "with_cells": true
30
- },
31
- "dataloader": {
32
- "batch_size": 1,
33
- "num_workers": 0
34
- }
35
- },
36
- "training": {
37
- "max_epoch": 1,
38
- "lr": 0.0001,
39
- "loss_alpha": 0.1,
40
- "patience": 10,
41
- "checkpoint_dir": "./weight",
42
- "seed": 0,
43
- "device": "cpu"
44
- },
45
- "inference": {
46
- "checkpoint_path": "./weight/best_model.pth",
47
- "output_dir": "./result/output",
48
- "batch_size": 1,
49
- "device": "cpu"
50
- }
51
  }
 
1
  {
2
+ "name": "EagleMeshTransformer",
3
+ "architecture": "GraphViT",
4
+ "state_size": 4,
5
+ "w_size": 32
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
6
  }