support pl module test step

BiomedSciAI / fuse-med-ml

A python framework accelerating ML based discovery in the medical field by encouraging code reuse. Batteries included :)

Apache License 2.0

134 stars 34 forks source link

Closed sivanravidos closed 6 months ago

sivanravidos commented 7 months ago

support running test step of pytorch lightning module, this is useful if we want to run predict on the test set and get also the metrics.