Skip to content
Discussion options

You must be logged in to vote

Hi @YingdiZhang, maybe you could use torch.nn.MSELoss, you can find a reference in this tutorial.

Hope it can help you, thanks!

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by YingdiZhang
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
2 participants