transformer encoder layer output deos not seem right!

[hoseinbehrad]

Hi. I was watching the video on 08. PyTorch Paper Replicating and also running code from 08_pytorch_paper_replicating_exercise_solutions.ipynb.
However when in the Excersice 1, 2.TransformerEncoderLayer section, the result are not generated I believe, or model is not trained.

I ran the solution notebook, but again the results are not the same.

The result from the video on 08. PyTorch Paper Replicating (time = 38:05)

The result from re-running the exact same code (from 08_pytorch_paper_replicating_exercise_solutions.ipynb)