diff options
Diffstat (limited to 'test/hpm.py')
| -rw-r--r-- | test/hpm.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/test/hpm.py b/test/hpm.py index a68337d..0aefbb8 100644 --- a/test/hpm.py +++ b/test/hpm.py @@ -1,6 +1,6 @@ import torch -from models import HorizontalPyramidMatching +from models.hpm import HorizontalPyramidMatching T, N, C, H, W = 15, 4, 256, 32, 16 |