第145行 transform_val.transforms.insert(0, Augmentation(auto_policy))
中插入的auto_policy
为二维数组,这就造成在augment.py文件中第55行 for name, pr, level in policy
中的policy
实际为一维数组,导致程序无法继续运行。
改为 transform_val.transforms.insert(0, Augmentation([auto_policy]))
可以解决这个问题。