IT 지식 창고
(albumentation) pytorch가 없다고 뜰 때
casim
2020. 9. 8. 18:22
제가 직접 github에 물어본 것으로, module ' albumentations' has no attribute 'pytorch'라는 error로 pytorch가 없다고 뜨는 경우에는 직접 import를 하면 됩니다.
import albumentations.pytorch
위와 같이 import를 시키니 잘 사용은 됩니다만, 왜 그런지 이유는 모르겠네요..
github.com/albumentations-team/albumentations/issues/692
module 'albumentations' has no attribute 'pytorch' · Issue #692 · albumentations-team/albumentations
🐛 Bug I want to use albumentations.pytorch library. but I get an error that there is no library. To Reproduce import albumentations import torch print(albumentations.__version__) '0.4.6' al...
github.com