@mfa hi, I am also trying to build AE for sentences with batch training. After creating  the custom dataset for sentences with padding and lengths parameters for each sentence, i am stuck how to move forward. Can you share your code?