 Tri Dao
|
c7f32a8409
[CrossEntropy] Support precomputed LSE
|
hai 3 meses |
|
Tri Dao
|
ec6d22143b
[CrossEntropy] Change ignored_index -> ignore_index
|
hai 7 meses |
 Curtis "Fjord" Hawthorne
|
d8aacc510c
return z_loss (#768)
|
hai 10 meses |
|
Tri Dao
|
08124c8f9c
[CrossEntropy] Implement logit_scale option
|
hai 1 ano |
|
Tri Dao
|
5400fdc4ac
[CE] Implement CrossEntropyLoss in Triton
|
hai 1 ano |
|
Tri Dao
|
f1a73d0740
Run isort and black on python files
|
hai 1 ano |
|
Tri Dao
|
c6ecd40a59
Tweak CrossEntropyLoss to take process_group in init
|
hai 1 ano |
|
Tri Dao
|
dff68c2b22
Add smoothing for CrossEntropyParallel, rename to CrossEntropyLoss
|
%!s(int64=2) %!d(string=hai) anos |
