baicenxiao

Thank you so much for the prompt response! I have set `ORI: False`, retrained and saved the best model. It gives better result, but still not able to achieve that...

Thanks! Let me try on other server. Did you use the same parameters (except for the `batch_size`) like below to train on SD7K dataset? I may try on SD7K as...

Appreciate your response @zinuoli. I just tried Google Colab to train the model on Jung dataset this time, the results are similar to the ones I got from my local...

Thanks for letting me know @zinuoli. The paper mentions that smoothL1 loss was used for training. But as I notice, in the code ([this line](https://github.com/CXH-Research/DocShadow-SD7K/blob/eb0789d46b1c7db79f4116ca65a1c3cfb58b5674/train.py#L47)), it is `MSEloss`. Did you...

Thanks @TobyRoseman for the response. It is still an issue even on the most recent version of iOS (17.1). Using the converted model for inference works fine, the problem happens...

Just submitted this issue using the [Feedback Assistant](https://developer.apple.com/bug-reporting/).