(prefix, mod)
| 1048 | |
| 1049 | |
| 1050 | def add_state_dict(prefix, mod): |
| 1051 | for k, v in mod.state_dict().items(): |
| 1052 | unet_state_dict[f"{prefix}.{k}"] = v |
| 1053 | |
| 1054 | |
| 1055 | add_state_dict("conv_in", conv_in) |
no test coverage detected
searching dependent graphs…