Setting `"train"` to `classes` argument of `LSUN()` kills the kernel on JupyterLab
Open
Nobody has claimed this yet.
- Dominant language
- Python
- Stars
- 17.9k
- Forks
- 7.3k
- Avg merge
- 1d 15h
- Merged PRs (30d)
- 13
Description
🐛 Describe the bug
Setting "train" to classes argument of LSUN() kills the kernel on JupyterLab as shown below:
*Memos:
- It also happens with
["bedroom_train"],["church_outdoor_train"],["classroom_train"],["kitchen_train"],["living_room_train"]and["restaurant_train"]forclassesargument. "test"doesn't work becausetest_lmdb.zip,test_train_lmdb.zipandtest_val_lmdb.zipcannot be unzipped as I reported the bug.- There is other error as I reported here.
from torchvision import datasets
train_data = datasets.LSUN(
root="C:/Users/<user>/dl-models/data/lsun",
classes="train" # Here
) # Error
print(len(train_data))
In addition, setting "val" to classes argument of LSUN() works properly on JupyterLab as shown below.
*Memos:
["bedroom_train_val"],["bridge_val"],["church_outdoor_val"],["classroom_val"],["conference_room_val"],["dining_room_val"],["kitchen_val"],["living_room_val"],["restaurant_val"]and["tower_val"]forclassesargument also work properly.testdoesn't work becausetest_lmdb.zip,test_train_lmdb.zipandtest_val_lmdb.zipcannot be unzipped as I reported the bug.
from torchvision import datasets
val_data = datasets.LSUN(
root="C:/Users/<user>/dl-models/data/lsun",
classes="val" # Here
)
print(len(val_data)) # 300
Versions
import torchvision
torchvision.__version__ # '0.20.1'
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Reproduce the failure with torchvision 0.20.1 by constructing LSUN with classes="train" in JupyterLab, then compare it with the working classes="val" case. The issue names no source file or test, so trace LSUN dataset initialization and add a regression check once the training dataset no longer kills the kernel.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- jupyter-notebook, python
- Domain
- computer-vision
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100