(k, v)
| 56 | ] |
| 57 | |
| 58 | def transform(k, v): |
| 59 | if any([k.startswith(name) for name in upsample_conv_names]): |
| 60 | out = ('generator.coarse_net.upsample_module.' + k, v) |
| 61 | elif any([k.startswith(name) for name in downsample_conv_names]): |
| 62 | out = ('generator.coarse_net.downsample_module.' + k, v) |
| 63 | else: |
| 64 | out = (k, v) |
| 65 | return out |
| 66 | |
| 67 | new_state_dict = OrderedDict([ |
| 68 | transform(k, v) |
no outgoing calls
no test coverage detected