Problems running UNet Industrial inference


I am running the notebook example of training and test UNet Industrial on DAGM dataset provided at DeepLearningExamples/blob/master/TensorFlow/Segmentation/UNet_Industrial.

I was able to train the model correctly, but when i am going to test it gives the error:

ValueError: Horovod has not been initialized; use hvd.init().

How do I solve that?


Please reach out to Issues · NVIDIA/DeepLearningExamples · GitHub to get better help.

Thank you.