Skip to content

Commit 17d80eb

Browse files
authored
Update main_test_table1.py
1 parent 1c1c71c commit 17d80eb

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

Diff for: main_test_table1.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,8 @@
1818
from utils import utils_logger
1919
from utils import utils_sisr as sr
2020
from utils import utils_image as util
21-
from models.network_usrnet import USRNet as net
21+
from models.network_usrnet import USRNet as net # for pytorch version <= 1.7.1
22+
# from models.network_usrnet_v1 import USRNet as net # for pytorch version >= 1.8.1
2223

2324

2425
'''

0 commit comments

Comments
 (0)