search

ATAboukhadra / THOR-Net

End-to-end Graformer-based Realistic Two Hands and Object Reconstruction with self-supervision
20 stars 2 forks source link

ImportError: cannot import name 'StreamWrapper' from 'torch.utils.data.datapipes.utils.common' #6

Open SUOXIAOXI opened 1 year ago

SUOXIAOXI commented 1 year ago

Hello, Thank you for your nice works! I have problems when trying to run the code: my environment: torch==1.10.1 ; torchvision==0.11.2 ; torchtext==0.11.1 ; torchdata==0.3.0a0; Where is my mistake？

ATAboukhadra commented 1 year ago

Hi,

I guess this is solvable by installing a newer version of TorchData like 0.4, however, you don't really need TorchData because we use it for creating an iterable dataset from tar files for the H2O dataset using an internal library.

I suggest installing a newer version of TorchData or removing it from the installation process and also removing all mentions of datapipes imports from the code. I guess also installing a newer PyTorch version >=1.12 could be useful. I hope this helps.

Kind regards, Ahmed

SUOXIAOXI commented 1 year ago

Thank you for your replying! I installed torch ==1.12.1 and solved the problem.