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...
Differentiating Through a Non-differentiable Function with `torch.autograd.Function`

this month hard (<1 hr) | autograd custom gradient function backprop

Implement a **custom `torch.autograd.Function`** for a non-differentiable operation, such as a custom quantization function. The `forward` method will perform the non-differentiable operation, and...