CARE 2D: Loss curves not displaying

HenriquesLab / ZeroCostDL4Mic

ZeroCostDL4Mic: A Google Colab based no-cost toolbox to explore Deep-Learning in Microscopy

MIT License

565 stars 129 forks source link

CARE 2D: Loss curves not displaying #335

Closed kaarjel closed 4 weeks ago

kaarjel commented 1 month ago

Hi,

I am using CARE 2D on Google Colab and am getting an error after the training has completed. I am also not able to visualize the loss curves of the training. Below is the error I am getting:

In the 'Inspection of loss function' cell, the plots are empty:

Hope you can help.

Cheers, Kaarjel

moore-cl commented 4 weeks ago

Hi Kaarjel,

I have had this error myself before, it seems to be because 'lr' is not defined and is referred to as 'learning rate' in line 25. By changing line 25 'learning rate' to 'learning_rate' and then line 27 'lr' to 'learning_rate' so they matched fixed the problem and my curves plot. The creators of the notebook can jump in here if thats not the correct fix.

kaarjel commented 4 weeks ago

Hi moore-cl,

That fix worked, thanks for your help and prompt reply!

Cheers, Kaarjel