MitoNet example / tutorial

[T3chbug]

Hi,
I am currently using MitoNet in napari, but my datasets are growing in size so want to automate the segmentation process using python (ideally with jupyter notebooks) instead of the point and click process in napari. I didn't find an minimal example of running the MitoNet models (with the same / similar parameters like in empanada-napari) in the documentation nor the repository of empanada-dl / empanada-napari.

I would be very happy to be pointed into the right direction or provided with some script to start from.
Thanks in advance,
Paul

PS: I already tried to load the models using torch.jit, causing a "not implemented" error related to cuda.