Update `SemanticSegmentationModule` - Githubissues

kaiko-ai / eva

Evaluation framework for oncology foundation models (FMs)

https://kaiko-ai.github.io/eva/

Apache License 2.0

66 stars 4 forks source link

Update `SemanticSegmentationModule` #636

Closed nkaenzig closed 1 month ago

nkaenzig commented 2 months ago

accept any nn.Module as decoder
allow forward pass with a single argument to the forward pass