sup3r.models.utilities.TrainingSession

sup3r.models.utilities.TrainingSession#

class TrainingSession(batch_handler, model, config=None, **kwargs)[source]#

Bases: object

Wrapper to gracefully exit batch handler thread during training, upon a keyboard interruption.

Parameters:

  • batch_handler (BatchHandler) – Batch iterator

  • model (Sup3rGan) – Gan model to run in new thread

  • config (TrainingConfig | None) – Training configuration for model.train().

  • **kwargs (dict) – Backwards-compatible keyword args for TrainingConfig.

Methods

run()

Wrap model.train().
run()[source]#

Wrap model.train().

Contents