Shortcuts

Source code for catalyst.dl.state.gan

from catalyst.core import State


[docs]class GanState(State): """ An object that is used to pass internal state during train/valid/infer in GAN Runners. """
[docs] def __init__(self, **kwargs): """ Args: @TODO: Docs. Contribution is welcome """ self.phase = None super().__init__(**kwargs)
__all__ = ["GanState"]