modlee.trainer module

Modlee Trainer.

class modlee.trainer.Trainer(*args, **kwargs)[source]

Bases: Trainer

Trainer that directs checkpoint files to the current run directory.