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"]