Function load_state_optimizer

PATH/core/utils.py:219–223

(optimizer, state, ginfo)

def load_state_optimizer(optimizer, state, ginfo):
    if ginfo.task_rank == 0:
        printlog(f'======= loading optimizer state for task {ginfo.task_id} ... =======')

    optimizer.load_state_dict(state)

def create_logger(name, log_file, level=logging.INFO):
    l = logging.getLogger(name)