Skip to content

Commit 4402c7b

Browse files
author
mumuychen
committed
[FIX]Fixed torch version
1 parent e30a75a commit 4402c7b

File tree

2 files changed

+2
-4
lines changed

2 files changed

+2
-4
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@ The repository is structured as follows:
2222
- weights/: Download the trained model weights file to the `weights` folder.
2323

2424
## Requirements
25-
- Torch == 0.3.1
26-
- Torchvision == 0.2.1
25+
- Torch >= 1.6.0
26+
- Torchvision >= 0.7.0
2727
- Python == 3.6
2828

2929
For training and inference, all supported algorithm dependencies are presented in [requirements.txt](requirements.txt).

evaluation.py

-2
Original file line numberDiff line numberDiff line change
@@ -336,7 +336,6 @@ def main(
336336
pair,
337337
matching,
338338
with_desc,
339-
warp_origin=warp_origin,
340339
)
341340
else:
342341
results = preprocess_match_pipeline(
@@ -352,7 +351,6 @@ def main(
352351
pair,
353352
matching,
354353
with_desc,
355-
warp_origin=warp_origin,
356354
)
357355
if 'icp' in config['matcher']['model']['name']:
358356
viz_path = name0 + '_' + name1

0 commit comments

Comments
 (0)