modelzoo.common.pytorch.run_cstorch_flow.fix_last_epoch#

modelzoo.common.pytorch.run_cstorch_flow.fix_last_epoch(lr_scheduler, global_step: int)[source]#

Generates a function that can be called repeatedly to fix the last epochs across all LR schedulers

Parameters

global_step – The global step value to use to fix up the last epoch values

previous

modelzoo.common.pytorch.run_cstorch_flow.compute_params_norm

next

modelzoo.common.pytorch.run_cstorch_flow.get_cluster_config