Skip to content

Instantly share code, notes, and snippets.

@justusschock

justusschock/rmse_plain_functional.py

Last active June 21, 2020 12:18
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save justusschock/d1a68b20964dbfa09b0bad5c87a322c1 to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save justusschock/d1a68b20964dbfa09b0bad5c87a322c1 to your computer and use it in GitHub Desktop.
Download ZIP
Plain functional example of a metric (RMSE)
Raw
rmse_plain_functional.py
import torch
def rmse(pred: torch.Tensor, target: torch.Tensor) -> torch.Tensor:
return torch.sqrt(torch.mean(torch.pow(pred-target, 2.0)))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment