sup3r.models.utilities.TrainingSession

sup3r.models.utilities.TrainingSession#

class TrainingSession(batch_handler, model, **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

  • **kwargs (dict) – Model keyword args

Methods

run()

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

Wrap model.train().