openfold.utils.logger

Classes

PerformanceLoggingCallback(log_file, ...[, ...])

Functions

is_main_process()

class PerformanceLoggingCallback(log_file, global_batch_size, warmup_steps=0, profile=False)

Bases: Callback

Parameters:

• warmup_steps (int)

• profile (bool)

do_step()

on_epoch_end(trainer, pl_module)

on_test_batch_start(trainer, pl_module, batch, batch_idx, dataloader_idx)

on_train_batch_start(trainer, pl_module, batch, batch_idx, dataloader_idx)

on_train_end(trainer, pl_module)

process_performance_stats(deltas)

is_main_process()