You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
keepdim is supposed to retain the dimensions of the input tensor passed. However here passing a CHW tensor into an augmentation results in a batch dim being added.
The above snippet shouldn't raise an assertion error. It shouldn't add a batch dim here
Environment
wget https://raw.githubusercontent.com/pytorch/pytorch/main/torch/utils/collect_env.py
# For security purposes, please check the contents of collect_env.py before running it.
python collect_env.py
kornia==0.7.2
PyTorch Version (e.g., 1.0): 2.2.0
OS (e.g., Linux): Linux
How you installed PyTorch (conda, pip, source): pip
Build command you used (if compiling from source): pip install -e .
Python version: 3.11
CUDA/cuDNN version:
GPU models and configuration:
Any other relevant information:
The text was updated successfully, but these errors were encountered:
a workaround, manually setting the keepdim after init the container... but I couldn't find at the time what was causing it, nor why the #2800 patch caused it.
...
aug_dict.keepdim=True
...
Basically, it is not being initialized and/or propagated correctly after the container is initialized. This only causes it to ignore trying to rebuild the correct dimensions before returning (by default keep dim is initialized as False).
Describe the bug
keepdim is supposed to retain the dimensions of the input tensor passed. However here passing a CHW tensor into an augmentation results in a batch dim being added.
Reproduction steps
Expected behavior
The above snippet shouldn't raise an assertion error. It shouldn't add a batch dim here
Environment
wget https://raw.githubusercontent.com/pytorch/pytorch/main/torch/utils/collect_env.py # For security purposes, please check the contents of collect_env.py before running it. python collect_env.py kornia==0.7.2
conda
,pip
, source): pippip install -e .
The text was updated successfully, but these errors were encountered: