O meu problema original era que eu queria que o meu modelo, o resultado de 0-1 para que eu possa mapa de volta para o meu categórica imagens de rótulos (Fluxo.jl restringir variáveis entre 0 e 1). Então, eu decidi adicionar uma função de ativação sigmóide da seguinte forma:
σ = 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
);
No entanto, agora o meu modelo só saídas 1.0. Alguma idéia do porquê ou se eu estou usando a função de ativação de errado?
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