from albumentations import torch as AT报错处理
问题
最近看kaggle大佬的代码,大家数据扩增(Augmentation)的时候都喜欢用albumentations,但是运行
from albumentations import torch as AT
会报错呀
解决办法
去git上面看了看albumentations的源代码,现在这句话应该写成
from albumentations import pytorch as AT
这样就好...
原创
2019-10-10 14:36:50 ·
1365 阅读 ·
1 评论