def METHOD_NAME(self):
    """To be implemented in subclasses.
    This method will be called before the first call to :fun:`train_step`,
    from the same process that :fun:`train_step` will be called from. Use
    this to initialize you solver and training hardware.
    """
    pass