search

dbolya / yolact

A simple, fully convolutional model for real-time instance segmentation.
MIT License
5k stars 1.33k forks source link

AttributeError: module 'torch.cuda' has no attribute '_UntypedStorage' #800

Open SAWRJJ opened 1 year ago

SAWRJJ commented 1 year ago

My torch vision is 1.12.0+cu116. I don't know this problem's reson. Have you guys encoutered this problem?

Priyashbhugra commented 1 year ago

@SAWRJJ try addding "--num_workers=0" to run script. It worked for me

SAWRJJ commented 1 year ago

@SAWRJJ try addding "--num_workers=0" to run script. It worked for me ok, I will try it. ty