search

jonbarron / robust_loss_pytorch

A pytorch port of google-research/google-research/robust_loss/
Apache License 2.0
656 stars 88 forks source link

GPU support #1

Closed HectorAnadon closed 5 years ago

HectorAnadon commented 5 years ago

Hello,

Thanks for the code. Great project. I have one question, do you have GPU support? I have only been able to run it in CPU.

Thanks for your help!

jonbarron commented 5 years ago

Oops, yes, you are correct. Sorry, this codebase is the first and only thing I've ever written in pytorch, and I'm still learning (in tensorflow this is a non-issue). I will make this GPU-compatible as soon as I can, hopefully within the next week.

jonbarron commented 5 years ago

Alright, it looks like GPU support is working. Please let me know if you have any trouble getting it to work.