When I just do_predict, I got KeyError of optimizer

[allenyummy]

When I just do_predict, I got KeyError of optimizer, like this:

Do you have any idea for the problem ?

[huminghao16]

The reason may be that the pre-trained model I uploaded does not contain the parameter for optimizer. So if you want to do prediction, you should mask out the following codes in the run file:

    global_step, global_epoch = 0, 1
    if os.path.isfile(save_path):
        checkpoint = torch.load(save_path)
        optimizer.load_state_dict(checkpoint['optimizer'])
        logger.info("Load optimizer from finetuned checkpoint: '{}' (step {}, epoch {})"
                    .format(save_path, checkpoint['step'], checkpoint['epoch']))
        global_step = checkpoint['step']
        global_epoch = checkpoint['epoch'] + 1