CoinCheung / pytorch-loss

label-smooth, amsoftmax, partial-fc, focal-loss, triplet-loss, lovasz-softmax. Maybe useful

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Any plan to give a tutorial?

seanzhuh opened this issue · comments

Hi, it's a great work! However, I often found myself stuck on how to write a custom cuda kernel / C++ extension, do you have any plan to write a tutorial or generously give some helpful links? I found most tutorials on internet including PyTorch's official document on custom cuda kernel is not that welcome to new beginners.

Writing tutorals about extend pytorch with cuda is beyond the scope of this repository, and, to be honest, it is also not in my todo list. But you can provide information about the problems you met here and let's see how we could make your extensions work.