snorkel.classification.LogManager¶
-
class
snorkel.classification.
LogManager
(n_batches_per_epoch, log_writer=None, checkpointer=None, **kwargs)[source]¶ Bases:
object
A class to manage logging during training progress.
- Parameters
n_batches_per_epoch (
int
) – Total number batches per epochlog_writer (
Optional
[LogWriter
]) –LogWriter
for current run logscheckpointer (
Optional
[Checkpointer
]) –Checkpointer
for current modelkwargs (
Any
) – Settings to update in LogManagerConfig
-
__init__
(n_batches_per_epoch, log_writer=None, checkpointer=None, **kwargs)[source]¶ Initialize self. See help(type(self)) for accurate signature.
- Return type
None
Methods
__init__
(n_batches_per_epoch[, log_writer, …])Initialize self.
cleanup
(model)Close the log writer and checkpointer if needed.
reset
()Reset counters.
Check if current counts trigger checkpointing.
Check if current counts trigger evaluation.
update
(batch_size)Update the count and total number.