cielal / torch-staintools Goto Github PK

Hello,

First of all, thank you for your work. I am writing you because I'm having a hard time isolating a problem : at some point during my training, the output of the on-the-fly normalization contains Nan Values. I was wondering if that can happen when the normalization reference (used to fit) is contained in the batch of images to normalize ? It doesn't always happen, which is very strange to me. It may be totally unrelated to my problem, but I need to get an answer to this lead

Have a good day

That should vastly improve the efficiency of Vahadane algorithm when on-the-fly normalization is necessary for multiple iterations, e.g., for multiple epochs of downstream deep learning model training.