HenriquesLab / ZeroCostDL4Mic

ZeroCostDL4Mic: A Google Colab based no-cost toolbox to explore Deep-Learning in Microscopy
MIT License
566 stars 130 forks source link

StarDist_2D_ZeroCostDL4Mic training super slow #217

Closed miinaojansivu closed 2 years ago

miinaojansivu commented 2 years ago

I am using StarDist_2D_ZeroCostDL4Mic to analyze my super-resolution images of lysosomes. I have done this for several datasets successfully in the past, the latest successful run being about a month ago. I have normally 5-15 training images which I have manually annotated in ImageJ (LOCI plugin). Now I'm trying to do exactly the same training for another similar dataset but for some reason the training is super slow, each epoch taking at least an hour, meaning that there's no way the training could ever be completed with this speed. All my training parameters are exactly the same as previously with the successful runs. I first noticed this problem around a week ago.

Normally the training takes up to 6h or so, depending on the amount of training data. This is what I'd expect to see here as well.

Desktop:

esgomezm commented 2 years ago

Hi @miinaojansivu

The conditions of Google Colab have changed and the old versions of the notebooks do not connect with GPUs anymore. If you will not use the model in Fiji, I would recommend you update the notebook. The performance will be the same.

Hope this works. I will close the issue but if you still have problems, please reopen it

Thanks!