TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix logitcrossentropy type

Browse Source
This commit is contained in:
Dhairya Gandhi 2019-01-03 19:45:38 +05:30
parent 65a41f2de6
commit 6a35fe07c9
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/layers/stateless.jl
View File

@ -10,7 +10,7 @@ end
@deprecate logloss(x, y) crossentropy(x, y) @deprecate logloss(x, y) crossentropy(x, y)
function logitcrossentropy(logŷ::AbstractVecOrMat, y::AbstractVecOrMat; weight = 1, efftype = eltype(ŷ)) function logitcrossentropy(logŷ::AbstractVecOrMat, y::AbstractVecOrMat; weight = 1, efftype = eltype(logŷ))
return -sum(y .* logsoftmax(logŷ) .* weight) / convert(efftype, size(y, 2)) return -sum(y .* logsoftmax(logŷ) .* weight) / convert(efftype, size(y, 2))
end end