(albumentation) pytorch가 없다고 뜰 때

제가 직접 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