fix methodmodel type restriction

Author: lorenzoh

src/methods/imagesegmentation.jl

@@ -203,7 +203,7 @@ end
 
 # ## Training interface
 
-function DLPipelines.methodmodel(method, backbone; kwargs...)
+function DLPipelines.methodmodel(method::ImageSegmentation, backbone; kwargs...)
     return UNetDynamic(
         backbone,
         (method.projections.sz..., 3, 1),

src/methods/singlekeypointregression.jl

@@ -160,7 +160,7 @@ end
 
 # Training interface
 
-DLPipelines.methodlossfn(method::SingleKeypointRegression) = Flux.mse
+DLPipelines.methodlossfn(::SingleKeypointRegression) = Flux.mse
 
 function DLPipelines.methodmodel(method::SingleKeypointRegression, backbone)
     h, w, ch, b = Flux.outdims(backbone, (method.projections.sz..., 3, 1))