Skip to content

Commit

Permalink
feat(torch): update torch to 1.13
Browse files Browse the repository at this point in the history
  • Loading branch information
Bycob authored and mergify[bot] committed Nov 10, 2022
1 parent 0f0bb62 commit 9c5da36
Show file tree
Hide file tree
Showing 9 changed files with 951 additions and 42 deletions.
6 changes: 3 additions & 3 deletions CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -772,15 +772,15 @@ if (USE_TORCH)

set(PYTORCH_PATCHES
${PYTORCH_PATCHES_PATH}/pytorch_15_compile.patch
${PYTORCH_PATCHES_PATH}/pytorch_19_new_logger.patch
${PYTORCH_PATCHES_PATH}/pytorch_113_new_logger.patch
${PYTORCH_PATCHES_PATH}/pytorch_19_use_new_logger.patch
)

message(STATUS "Configuring libtorch")
add_definitions(-DUSE_TORCH)

if (NOT TORCH_LOCATION)
set(PYTORCH_COMMIT v1.12.1)
set(PYTORCH_COMMIT v1.13.0)
set(PYTORCH_COMPLETE ${CMAKE_BINARY_DIR}/CMakeFiles/pytorch-complete)

if(USE_TORCH_CPU_ONLY)
Expand Down Expand Up @@ -820,7 +820,7 @@ if (USE_TORCH)

# TORCH VISION
message(STATUS "Configuring Pytorch Vision")
set(PYTORCH_VISION_COMMIT "v0.13.1") # 0.13.0 & torch 1.12.0
set(PYTORCH_VISION_COMMIT "v0.14.0") # 0.14.0 & torch 1.13.0
set(PYTORCH_VISION_PATCHES_PATH ${CMAKE_BINARY_DIR}/patches/pytorch/vision)
# set(PYTORCH_VISION_PATCHES
# ${PYTORCH_VISION_PATCHES_PATH}/xxx.patch # add patches here
Expand Down
Loading

0 comments on commit 9c5da36

Please sign in to comment.