Allow batch size to be set on the extension

DBM-MCF / qupath-extension-efficientv2unet

A QuPath extension for using Efficient V2 UNet

Apache License 2.0

1 stars 0 forks source link

Closed lacan closed 3 months ago

lacan commented 5 months ago

Hi, When using the default settings on a 8GB GPU, I get an OOM error

Exposing batch size would allow for smaller GPUs to run.

loicsauteur commented 3 months ago

Done