(config_path)
| 39 | |
| 40 | |
| 41 | def create_model(config_path): |
| 42 | config = OmegaConf.load(config_path) |
| 43 | model = instantiate_from_config(config.model).cpu() |
| 44 | print(f'Loaded model config from [{config_path}]') |
| 45 | return model |
| 46 | |
| 47 | def compare_weights(state_dict, layer1_name, layer2_name): |
| 48 | if layer1_name not in state_dict: |
no test coverage detected