clean

molo32 · Feb 12, 2020 · ccd2c36 · ccd2c36
1 parent 93a7cf5
commit ccd2c36
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/PWCNet/PWCNet.py b/PWCNet/PWCNet.py
@@ -12,8 +12,7 @@
 os.environ['PYTHON_EGG_CACHE'] = 'tmp/' # a writable directory 
 #from .correlation_package.modules.corr import Correlation
 # from PWCNet.correlation_package_pytorch0_4.correlation import Correlation #pytorch0.4 version
-# from PWCNet.correlation_package_pytorch1_0.correlation import Correlation #pytorch0.4 version
-from PWCNet.correlation_package_pytorch1_2.correlation import Correlation #pytorch0.4 version
+from PWCNet.correlation_package_pytorch1_0.correlation import Correlation #pytorch0.4 version
 
 import numpy as np