Round trip torch.save
/ torch.load
not working for Hyperplane
#2126
Labels
help wanted
Extra attention is needed
torch.save
/ torch.load
not working for Hyperplane
#2126
Describe the bug
Crashing when trying to save and reload a
Hyperplane
usingtorch
:Reproduction steps
$ python3 -m venv venv $ source venv/bin/activate (venv) $ pip install kornia (venv) $ pip install numpy (venv) $ python save_hyperplane.py
Content of save_hyperplane.py
Expected behavior
Hyperplane is able to be saved and reloaded with
torch.save
&torch.load
Environment
Collecting environment information... PyTorch version: 1.13.1+cu117 Is debug build: False CUDA used to build PyTorch: 11.7 ROCM used to build PyTorch: N/A OS: Ubuntu 20.04.5 LTS (x86_64) GCC version: (Ubuntu 9.4.0-1ubuntu1~20.04.1) 9.4.0 Clang version: Could not collect CMake version: version 3.16.3 Libc version: glibc-2.31 Python version: 3.8.10 (default, Nov 14 2022, 12:59:47) [GCC 9.4.0] (64-bit runtime) Python platform: Linux-5.15.0-56-generic-x86_64-with-glibc2.29 Is CUDA available: True CUDA runtime version: Could not collect CUDA_MODULE_LOADING set to: LAZY GPU models and configuration: GPU 0: NVIDIA GeForce RTX 3080 Laptop GPU Nvidia driver version: 470.141.03 cuDNN version: Could not collect HIP runtime version: N/A MIOpen runtime version: N/A Is XNNPACK available: True Versions of relevant libraries: [pip3] numpy==1.24.1 [pip3] torch==1.13.1 [conda] Could not collect
Additional context
No response
The text was updated successfully, but these errors were encountered: