stevenbucaille commited on
Commit
0f5fa28
1 Parent(s): 492409b
Files changed (2) hide show
  1. config.json +3 -0
  2. model.safetensors +2 -2
config.json CHANGED
@@ -3,6 +3,7 @@
3
  "architectures": [
4
  "SuperGlueForKeypointMatching"
5
  ],
 
6
  "descriptor_dim": 256,
7
  "gnn_layers_types": [
8
  "self",
@@ -24,6 +25,7 @@
24
  "self",
25
  "cross"
26
  ],
 
27
  "initializer_range": 0.02,
28
  "keypoint_detector_config": {
29
  "model_type": "superpoint"
@@ -36,6 +38,7 @@
36
  ],
37
  "matching_threshold": 0.2,
38
  "model_type": "superglue",
 
39
  "num_heads": 4,
40
  "sinkhorn_iterations": 100,
41
  "torch_dtype": "float32",
 
3
  "architectures": [
4
  "SuperGlueForKeypointMatching"
5
  ],
6
+ "attention_probs_dropout_prob": 0,
7
  "descriptor_dim": 256,
8
  "gnn_layers_types": [
9
  "self",
 
25
  "self",
26
  "cross"
27
  ],
28
+ "hidden_size": 256,
29
  "initializer_range": 0.02,
30
  "keypoint_detector_config": {
31
  "model_type": "superpoint"
 
38
  ],
39
  "matching_threshold": 0.2,
40
  "model_type": "superglue",
41
+ "num_attention_heads": 4,
42
  "num_heads": 4,
43
  "sinkhorn_iterations": 100,
44
  "torch_dtype": "float32",
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:44126312455dc3bab8a30cb1761ee65f52028f952e97822819921c8a062a01c3
3
- size 53412728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:70cad80b64ab9e0e57c0a7d354dd8ddda4b5cba56b594b9c4d0650c792bc69c9
3
+ size 53413232