search

binli123 / dsmil-wsi

DSMIL: Dual-stream multiple instance learning networks for tumor detection in Whole Slide Image
MIT License
332 stars 84 forks source link

What is aggregator.pth? #51

Closed jianan-zhang closed 1 year ago

jianan-zhang commented 1 year ago

Thanks for sharing this amazing work. I wonder what is the 'aggregator.pth' for the dsmil model. Is it trained on some specific dataset? How can I get it on my own dataset? Looking forward to your reply.

Best Wishes!

binli123 commented 1 year ago

The ‘aggregator.pth' is the provided pretrained weights for the dsmil model for testing. If you trained your own model, your weights will be saved in the folder 'weights/{date}/*.pth'. You can then load it for testing.

jianan-zhang commented 1 year ago

Thank you so much.