After solving the dataset malloc problem, another problem of "NotImplementedError"

[paperplane03]

This is the picture of the bug. Can you share official torchvision version, maybe it can help me solve the porblem, since I find the problem is related to:

This is run on linux (ubuntu22.04, cuda:12.4, memory: 103G, and pip list as below):
Package Version Editable project location

---

absl-py 2.1.0
aiobotocore 2.17.0
aiohappyeyeballs 2.4.4
aiohttp 3.11.11
aioitertools 0.12.0
aiosignal 1.3.2
anaconda-anon-usage 0.4.4
annotated-types 0.7.0
archspec 0.2.3
asciitree 0.3.3
atomicwrites 1.4.1
attrs 25.1.0
boltons 23.0.0
boto3 1.35.81
botocore 1.35.93
Brotli 1.0.9
cachetools 5.5.1
cellmap-data 2025.2.4.1150
cellmap-schemas 0.8.0
cellmap-segmentation-challenge 0.0.1 /home/yhzhong/projects/CellSeg/cellmap-segmentation-challenge
cellpose 3.1.0
certifi 2024.7.4
cffi 1.16.0
charset-normalizer 3.3.2
click 8.1.8
cloudpickle 3.1.1
conda 24.7.1
conda-content-trust 0.2.0
conda-libmamba-solver 24.7.0
conda-package-handling 2.3.0
conda_package_streaming 0.10.0
contourpy 1.3.1
cryptography 42.0.5
cycler 0.12.1
dask 2025.1.0
Deprecated 1.2.18
distro 1.9.0
eval-type-backport 0.1.3
fasteners 0.19
fastremap 1.15.0
filelock 3.17.0
flexcache 0.3
flexparser 0.4
fonttools 4.55.8
frozendict 2.4.2
frozenlist 1.5.0
fsspec 2025.2.0
google-apitools 0.5.32
google-auth 2.38.0
grpcio 1.70.0
httplib2 0.22.0
idna 3.7
imagecodecs 2024.12.30
imageio 2.37.0
Jinja2 3.1.5
jmespath 1.0.1
joblib 1.4.2
jsonpatch 1.33
jsonpointer 2.1
kiwisolver 1.4.8
lazy_loader 0.4
libmambapy 1.5.8
llvmlite 0.44.0
locket 1.0.0
Markdown 3.7
markdown-it-py 3.0.0
MarkupSafe 3.0.2
matplotlib 3.10.0
mdurl 0.1.2
menuinst 2.1.2
ml_collections 1.0.0
ml_dtypes 0.5.1
mpmath 1.3.0
multidict 6.1.0
natsort 8.4.0
networkx 3.4.2
neuroglancer 2.40.1
numba 0.61.0
numcodecs 0.15.0
numpy 2.0.2
nvidia-cublas-cu12 12.4.5.8
nvidia-cuda-cupti-cu12 12.4.127
nvidia-cuda-nvrtc-cu12 12.4.127
nvidia-cuda-runtime-cu12 12.4.127
nvidia-cudnn-cu12 9.1.0.70
nvidia-cufft-cu12 11.2.1.3
nvidia-curand-cu12 10.3.5.147
nvidia-cusolver-cu12 11.6.1.9
nvidia-cusparse-cu12 12.3.1.170
nvidia-cusparselt-cu12 0.6.2
nvidia-nccl-cu12 2.21.5
nvidia-nvjitlink-cu12 12.4.127
nvidia-nvtx-cu12 12.4.127
oauth2client 4.1.3
opencv-python-headless 4.11.0.86
packaging 24.1
pandas 2.2.3
partd 1.4.2
pillow 11.1.0
Pint 0.24.4
pip 24.2
platformdirs 3.10.0
pluggy 1.0.0
propcache 0.2.1
protobuf 5.29.3
pyasn1 0.6.1
pyasn1_modules 0.4.1
pycosat 0.6.6
pycparser 2.21
pydantic 2.10.6
pydantic_core 2.27.2
pydantic-ome-ngff 0.6.0
pydantic-zarr 0.7.0
Pygments 2.19.1
pyparsing 3.2.1
PySocks 1.7.1
python-dateutil 2.9.0.post0
python-dotenv 1.0.1
pytz 2025.1
PyYAML 6.0.2
requests 2.32.3
rich 13.9.4
roifile 2024.9.15
rsa 4.9
ruamel.yaml 0.17.21
s3fs 2025.2.0
s3transfer 0.10.4
scikit-image 0.25.1
scikit-learn 1.6.1
scipy 1.15.1
setuptools 72.1.0
six 1.17.0
structlog 25.1.0
sympy 1.13.1
tensorboard 2.18.0
tensorboard-data-server 0.7.2
tensorboardX 2.6.2.2
tensorstore 0.1.71
threadpoolctl 3.5.0
tifffile 2025.1.10
toolz 1.0.0
torch 2.6.0
torchvision 0.21.0
tornado 6.4.2
tqdm 4.66.4
triton 3.2.0
truststore 0.8.0
typing_extensions 4.12.2
tzdata 2025.1
universal_pathlib 0.2.6
urllib3 2.2.2
Werkzeug 3.1.3
wheel 0.43.0
wrapt 1.17.2
xarray 2025.1.2
xarray-ome-ngff 3.1.1
xarray-tensorstore 0.1.5
yarl 1.18.3
zarr 2.18.4
zstandard 0.22.0

[paperplane03]

The solution is to change _transform to tranform.

[rhoadesScholar]

@paperplane03 interesting. I have been using torchvision==0.20.1 without a problem. This must be a version issue, but it sounds like it is easy to fix. Thank you! Would you like to make a PR to get credit for the fix?