Caplin43 commited on
Commit
91ad132
·
verified ·
1 Parent(s): 4eea172

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +9 -0
config.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_type": "transformer",
3
+ "architecture": "InteractionNet",
4
+ "hidden_size": 512,
5
+ "num_attention_heads": 8,
6
+ "num_hidden_layers": 6,
7
+ "activation": "relu",
8
+ "framework": "pytorch"
9
+ }