All Katas - ML Katas

by: newest upvotes saves

Implementing a Custom Loss Function with `torch.autograd`

this month medium (<30 mins) | regression loss custom function huber

Create a **custom loss function** that inherits from `torch.nn.Module` and performs a non-standard calculation. For example, a custom Huber loss. This loss is less sensitive to outliers than Mean...