Method default_config

axlearn/common/evaler_test.py:81–84

(cls)

    @classmethod
    def default_config(cls):
        cfg = super().default_config()
        cfg.shape = _EXAMPLE_SHAPE
        return cfg

    def batches(self, it: tf.data.Iterator):
        yield from it