Skip to content

Commit bed581e

Browse files
committed
Cleanup torch-tensor-loading example.
1 parent 6b33353 commit bed581e

7 files changed

Lines changed: 4 additions & 5 deletions

File tree

examples/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,4 @@
22

33
add_subdirectory(my_example)
44

5-
add_subdirectory(torch_model_loading)
5+
add_subdirectory(torch-tensor-loading)

examples/torch_model_loading/CMakeLists.txt renamed to examples/torch-tensor-loading/CMakeLists.txt

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -19,9 +19,9 @@ target_link_options(TorchLoad
1919

2020
add_custom_command(TARGET TorchLoad POST_BUILD
2121
COMMAND ${CMAKE_COMMAND} -E copy_directory
22-
"${CMAKE_CURRENT_SOURCE_DIR}/models"
23-
"${CMAKE_BINARY_DIR}/TorchLoad_models"
24-
COMMENT "Copying model folder"
22+
"${CMAKE_CURRENT_SOURCE_DIR}/tensors"
23+
"${CMAKE_BINARY_DIR}/examples/TorchLoad/tensors"
24+
COMMENT "Copying tensor folder"
2525
)
2626

2727
set_target_properties(TorchLoad PROPERTIES
File renamed without changes.
File renamed without changes.

examples/torch_model_loading/torch_load.chpl renamed to examples/torch-tensor-loading/torch_load.chpl

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,6 @@ proc main(args: [] string) {
1414
var batchedImage = ndarray.loadPyTorchTensor(3,args[1],real(32)).unsqueeze(0);
1515
writeln("Batched image: ", batchedImage.shape);
1616

17-
// batchedImage = batchedImage.resize(224,224);
1817
writeln("Batched image resized: ", batchedImage.shape);
1918

2019
image = batchedImage.squeeze(3).imageNetNormalize();
-978 KB
Binary file not shown.
-1.91 MB
Binary file not shown.

0 commit comments

Comments
 (0)