Mon problème était que je voulais que mon modèle à sortie 0-1 afin que je puisse la carte de retour à mon catégorique images des étiquettes (de Flux.jl restreindre les variables entre 0 et 1). J'ai donc décidé d'ajouter une fonction d'activation sigmoïde comme suit:
σ = 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
);
Cependant, maintenant, mon modèle n'sorties 1.0. Toutes les idées pour lesquelles ou si je suis à l'aide de la fonction d'activation de 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