Skip to content

Commit ddcdeac

Browse files
committed
add template to ensemble config
1 parent 3ace30a commit ddcdeac

File tree

1 file changed

+13
-0
lines changed

1 file changed

+13
-0
lines changed
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,15 @@
11
class_path: chebai.models.ensemble.ChebiEnsemble
22
init_args:
3+
model_configs: {
4+
"model_1_name": {
5+
"ckpt_path": "path/to/your/model_1/checkpoint.ckpt",
6+
"class_path": "path/to/your/model_1/class",
7+
"labels_path": "path/to/your/model_1/classes.json",
8+
},
9+
10+
"model_2_name": {
11+
"ckpt_path": "path/to/your/model_2/checkpoint.ckpt",
12+
"class_path": "path/to/your/model_2/class",
13+
"labels_path": "path/to/your/model_2/classes.json",
14+
},
15+
}

0 commit comments

Comments
 (0)