How to extract the baseline model?

[chenyucheng0221]

Thank you for your awesome work.

Could you kindly help point out which part I can modify to extract the baseline network?

Thanks