We would still need a code snippet to further debug the issue.
I would recommend to stick to the tutorial and use the train transformation. Note that the normalization will be done in side the model as seen here.