To control the network configuration

datvuthanh / HybridNets

HybridNets: End-to-End Perception Network

MIT License

585 stars 120 forks source link

To control the network configuration #22

Closed rose-jinyang closed 2 years ago

rose-jinyang commented 2 years ago

Hello How are you? I want to train a model for ONLY object detection except segmentation. Could u provide a way to control this in .yaml file? Thanks.

datvuthanh commented 2 years ago

You can read our strategy in README. Basically, you can add --freeze_seg True command before training.