Ana Sanchez commited on
Commit
04aff2f
1 Parent(s): 58eeba9

Add model config

Browse files
Files changed (1) hide show
  1. RN50.json +15 -0
RN50.json ADDED
@@ -0,0 +1,15 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "embed_dim": 512,
3
+ "vision_block": "bottleneck",
4
+ "input_channels": 5,
5
+ "vision_layers": [
6
+ 3,
7
+ 4,
8
+ 6,
9
+ 3
10
+ ],
11
+ "vision_width": 64,
12
+ "input_size": 1024,
13
+ "molecule_layers": 4,
14
+ "hidden_dim": 1024
15
+ }