Spaces:
Sleeping
Sleeping
Ana Sanchez
commited on
Commit
·
04aff2f
1
Parent(s):
58eeba9
Add model config
Browse files
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 |
+
}
|