InfoGain Loss layerfor nvcaffe - request

NVIDIA / DIGITS

Deep Learning GPU Training System

https://developer.nvidia.com/digits

BSD 3-Clause "New" or "Revised" License

4.12k stars 1.38k forks source link

InfoGain Loss layerfor nvcaffe - request #1694

Open lamdawr opened 7 years ago

lamdawr commented 7 years ago

I find the InfoGain loss layer that is now available in caffe (https://github.com/BVLC/caffe/pull/3855 ) very useful for my multi-class classification application with highly imbalanced data distribution. It would be wonderful to have InfoGain loss layer in the nvcaffe so I can get the best out of the GPUs and DIGITS. Can someone help me in implementing that in nvcaffe or a new feature request for nvcaffe. Thanks in advance for the help.

druedaplata commented 7 years ago

Any news on this thread? I am interested as well.