(self, concat_keys=("lr",), reshuffle_patch_size=None,
low_scale_config=None, low_scale_key=None, *args, **kwargs)
| 1747 | condition on low-res image (and optionally on some spatial noise augmentation) |
| 1748 | """ |
| 1749 | def __init__(self, concat_keys=("lr",), reshuffle_patch_size=None, |
| 1750 | low_scale_config=None, low_scale_key=None, *args, **kwargs): |
| 1751 | super().__init__(concat_keys=concat_keys, *args, **kwargs) |
| 1752 | self.reshuffle_patch_size = reshuffle_patch_size |
| 1753 | self.low_scale_model = None |
| 1754 | if low_scale_config is not None: |
| 1755 | print("Initializing a low-scale model") |
| 1756 | assert exists(low_scale_key) |
| 1757 | self.instantiate_low_stage(low_scale_config) |
| 1758 | self.low_scale_key = low_scale_key |
| 1759 | |
| 1760 | def instantiate_low_stage(self, config): |
| 1761 | model = instantiate_from_config(config) |
nothing calls this directly
no test coverage detected