Skip to content

Commit dbafe3a

Browse files
benizmergify[bot]
authored andcommitted
chore: bump torchvision to 0.8.2
1 parent 38b1d07 commit dbafe3a

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

CMakeLists.txt

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -782,7 +782,8 @@ if (USE_TORCH)
782782
# TORCH VISION
783783
message(STATUS "Configuring Pytorch Vision")
784784
# set(PYTORCH_VISION_COMMIT "v0.6.0") # 0.6 & torch 1.5
785-
set(PYTORCH_VISION_COMMIT "v0.7.0") # 0.7 & torch 1.6
785+
#set(PYTORCH_VISION_COMMIT "v0.7.0") # 0.7 & torch 1.6
786+
set(PYTORCH_VISION_COMMIT "v0.8.2") # 0.8.2 & torch 1.7.1
786787
set(TORCHVISION_LOCATION "${CMAKE_BINARY_DIR}/pytorch_vision/src/pytorch_vision-install")
787788
ExternalProject_Add(
788789
pytorch_vision

0 commit comments

Comments
 (0)