(cls)
| 23 | ] |
| 24 | |
| 25 | def register_norm_module(cls): |
| 26 | NORM_MODULES.append(cls) |
| 27 | return cls |
| 28 | |
| 29 | def align_and_update_state_dicts(model_state_dict, ckpt_state_dict): |
| 30 | model_keys = sorted(model_state_dict.keys()) |
nothing calls this directly
no outgoing calls
no test coverage detected