Visualise and Fine-tuning model ATSS

[bzha5848]

What is the problem this feature will solve?

fine-tuning the custom dataset and visualise the model network

What is the feature you are proposing to solve the problem?

I need to freeze everything except the head of the model, and since currently the pre-trained model only trained 12 epoch for coco dataset, I also need either pre-trained model with 300 epoch on coco or pascal voc or do it by myself :(

What alternatives have you considered?

No response