Open rishabh316 opened 1 year ago
Dear @rishabh316 ,
Thank you for your inquiry.
But I am very sorry that there is currently no auto-grad op defined for our CRF module, and therefore it cannot be applied to model training. It is suggested that our CRF module can be used as a standalone processing or as a plug-in module for the well-trained model.
I hope my reply can be accepted. And could you kindly cite our relevant research if our code is helpful?
Best regards,
Libin
I applied this model on my dataset of images, converted them into arrays and feed them to the model. The model gets compiled after that. I am also getting the model summary, but when i try to fit the model using model.fit as : unet_model.fit(x, y, epochs=10) I get the following error:
StagingError: in user code:
Please provide the suitable solution for this problem asap. Notify me if any particular file or code is required.