Method get_eager_model

examples/models/edsr/model.py:20–24

(self)

        pass

    def get_eager_model(self) -> torch.nn.Module:
        logging.info("Loading edsr model")
        m = edsr_r16f64(2, True)
        logging.info("Loaded edsr model")
        return m

    def get_example_inputs(self):
        tensor_size = (1, 3, 224, 224)