search

bermanmaxim / LovaszSoftmax

Code for the Lovász-Softmax loss (CVPR 2018)
http://bmax.im/LovaszSoftmax
MIT License
1.38k stars 269 forks source link

added Lovasz-loss using pytorch tensors #37

Open gguzzy opened 7 months ago

gguzzy commented 7 months ago
  1. Introduced pytorch tensors to Loss, taking advantage of GPUs in torch.
  2. Added new example test using pytorch tensors
gguzzy commented 7 months ago

If you want, I can still arrange in a better way the code, aiming at realizing a custom torch loss, would you like to update the code together or can i progress with it? It should still work, but I wanted to be sure on that, what you think? I think it would be really useful for many tasks, that is why I am working on it. @bermanmaxim

Thanks