def load()

in utils/checkpoint.py [0:0]

• 22 lines of code
• 6 McCabe index (conditional complexity)

def load(args, model, optimizer, logger, scheduler):
    ep_init = 0
    if args.checkpoint != "" and os.path.exists(args.checkpoint):
        try:
            ep_init = load_checkpoint(
                args, args.checkpoint, model, optimizer, logger, scheduler
            )
        except Exception as e:
            print(f"load failed: {e}")
            # try the backup checkpoint
            if os.path.exists(args.checkpoint + ".bak"):
                try:
                    ep_init = load_checkpoint(
                        args,
                        args.checkpoint + ".bak",
                        model,
                        optimizer,
                        logger,
                        scheduler,
                    )
                except Exception as e:
                    print(f"backup load failed: {e}")
    return ep_init