The test loss is huge, which doesn’t seem right.
Are these outputs created in these lines of code 4?
PS: You can post code snippets by wrapping them into three backticks ```, which makes debugging easier.