ModelSummary can't show the mode of model whitch is frozen

[YChienHung]

Description & Motivation

Current ModelSummary model can only show the mode of module in two state, either train or eval, but we want to know if the module is frozen.

Pitch

No response

Alternatives

No response

Additional context

No response

cc @lantiga @Borda

[Borda]

Well, but that is not the third stage, right?
you can still have frozen model and being eval True/False

[YChienHung]

Well, but that is not the third stage, right? you can still have frozen model and being eval True/False

Yes, it is just useful for multi stage training environment, such as controlnet.