Mi problema original era que yo quería mi modelo a la salida 0-1 para que yo pueda mapa de nuevo a mi categórico etiquetas de las imágenes (de Flujo.jl restringir las variables entre 0 y 1). Así que he decidido añadir una función de activación sigmoide de la siguiente manera:
σ = 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
);
Sin embargo, ahora mi modelo solo salidas 1.0. Alguna idea de por qué o si estoy utilizando la función de activación mal?
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