search

HHHedo / IBMIL

CVPR 2023 Highlight
72 stars 9 forks source link

_'milnet' is not defined_ in _train_tcga.py_ #1

Closed jingweitan closed 1 year ago

jingweitan commented 1 year ago

Hi, I am trying to implement your code. But, I found that in the _traintcga.py , there is an error occured 'milnet' is not defined . May I know where can I get the milnet model? Thanks in advance.

HHHedo commented 1 year ago

Hi, I am trying to implement your code. But, I found that in the _traintcga.py , there is an error occured 'milnet' is not defined . May I know where can I get the milnet model? Thanks in advance.

We have fixed it. Please refer to: https://github.com/HHHedo/IBMIL/blob/f7c776c7cf194d2e74961c425cac625096699513/train_tcga.py#L281-L289

jingweitan commented 1 year ago

Thanks a lot!