How to Freeze the Whole Backbone

[weikai520]

What is the problem this feature will solve?

I think there are parameters that can freeze a layer or - 1 means that all layers are not frozen, and the whole backbone network is not frozen

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

I think there are parameters that can freeze a layer or - 1 means that all layers are not frozen, and the whole backbone network is not frozen

What alternatives have you considered?

No response

[RangeKing]

For the P5 model, setting frozen_stages=4 can freeze the whole backbone.
You can refer to the source code,

mmyolo/mmyolo/models/backbones/base_backbone.py

Lines 197 to 205 in c037ac8

	def _freeze_stages(self):
	"""Freeze the parameters of the specified stage so that they are no
	longer updated."""
	if self.frozen_stages >= 0:
	for i in range(self.frozen_stages + 1):
	m = getattr(self, self.layers[i])
	m.eval()
	for param in m.parameters():
	param.requires_grad = False