cannot import name 'data_augmentation' from 'data'

[Magic-Pony]

Hi, I'm trying to test the trained model with the provided checkpoint. But there is an import error indicates that I cannot import name 'data_augmentation' from 'data'. I checked the data folder and find there is not any data_augmentation.py script. Would you consider providing this script lately? Thank you.

[TIMESTICKING]

same here