Thanks for your work, I'm running the Shape Retrieval task but I get the following error:  and I get this error (autoencoder/train) 