Giters

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

TPU support

talhaanwarch opened this issue · comments

commented

Are these functions especially version v2 and v3 are supported on GPU only? or they are supported on tpu also?

commented

v1 and v2 uses pytorch native operators, which should work with tpu. As for v3 operators, they are specially written with cuda and can only work with gpu.