How to extract the baseline model?

[Leepoet]

Thank you for your awesome work.

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

Thanks