INNER CODE UNIT · Python

milestones

QizhiPei/FABind · FABind/fabind/main_fabind.py:286

    milestones=[args.warmup_epochs*len(train_loader)],
)

(
    model,
    optimizer,
    scheduler,
    train_loader,
) = accelerator.prepare(
    model, optimizer, scheduler, train_loader,
)

output_last_epoch_dir = f"{pre}/models/epoch_last"
if os.path.exists(output_last_epoch_dir) and os.path.exists(os.path.join(output_last_epoch_dir, "pytorch_model.bin")):
    # ckpt = os.path.join(args.resultFolder, args.exp_name, 'models', "epoch_last.pt")
    # model_ckpt, opt_ckpt, model_args, last_epoch = torch.load(ckpt)
    # model.load_state_dict(model_ckpt, strict=True)
    # optimizer.load_state_dict(opt_ckpt)

View source record →