stevenbucaille commited on
Commit
897c5eb
·
verified ·
1 Parent(s): 0f18f9e
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:c3449b17d1dd63a7a146ba0c0ce4bb26232e801ff69d173a45afb4e17071e22c
3
- size 53412728
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc2914273b209ab18134e6ca63ceb3d69290ef3f0e19b238c1f5b7f4d544360d
3
+ size 53413232