What kind of library for the data augmentation are you using? It sounds strange that your local SSD would be used as I don’t know why it would be necessary and it could also decrease the performance if you need to write/read from the disk while transforming data.