Můj původní problém byl, že jsem chtěl svůj model pouze výstup 0-1 tak můžu mapu zpět na můj kategorický obrázky štítky (Flux.jl omezit proměnné mezi 0 a 1). Tak jsem se rozhodl přidat sigmoid aktivační funkce takto:
σ = sigmoid
model = Chain(
resnet[1:end-2],
Dense(2048, 1000),
Dense(1000, 256),
Dense(256, 2, σ), # we get 2048 features out, and we have 2 classes
);
Nicméně, teď můj model pouze výstupy 1.0. Nějaké nápady, proč, nebo jestli jsem pomocí aktivace funkce špatně?
MethodError: no method matching similar(::Float32, ::Type{Float32}) Closest candidates are: similar(!Matched::Union{LinearAlgebra.Adjoint{T, var"#s814"}, LinearAlgebra.Transpose{T, var"#s814"}} where {T, var"#s814"<:(AbstractVector{T} where T)}, ::Type{T}) where T at /Users/julia/buildbot/worker/package_macos64/build/usr/share/julia/stdlib/v1.6/LinearAlgebra/src/adjtrans.jl