matsengrp / torchdms

Analyze deep mutational scanning data with PyTorch
https://matsengrp.github.io/torchdms/
2 stars 0 forks source link

Add huber loss #139

Closed zorian15 closed 2 years ago

zorian15 commented 3 years ago

Last meeting we discussed giving RMSE & Huber Loss a go on CGG data.

We currently don't support Huber loss in torchdms, but PyTorch has an implementation that will be quick and easy to integrate.

I'll begin by just using the default parameters for delta, which results in a SmoothL1Loss -- but will add some capabilities to toggle this after things are up and running.