79fc09d66d3f4736b9fb4f5756a78725719f3808,models.py,,setup,#Any#,13

Before Change

        // check if all necessary files exist 
        assert os.path.isdir(opt.start_from)," %s must be a a path" % opt.start_from
        assert os.path.isfile(os.path.join(opt.start_from,"infos_"+opt.id+".pkl")),"infos.pkl file does not exist in path %s"%opt.start_from
        return torch.load(os.path.join(opt.start_from, "model.pth"))

    if opt.caption_model == "show_tell":
        return ShowTellModel(opt)
    // elif opt.caption_model == "attention":

After Change

def setup(opt):
    
    if opt.caption_model == "show_tell":
        model = ShowTellModel(opt)
    // elif opt.caption_model == "attention":
    //     return AttentionModel(opt)
    // elif opt.caption_model == "show_attend_tell":
    //     return ShowAttendTellModel(opt)
    // elif opt.caption_model == "show_attend_tell_new":
    //     return ShowAttendTellModel_new(opt)
    // elif opt.caption_model == "test_att":
    //     return TestAttentionModel(opt)
    else:
        raise Exception("Caption model not supported: {}".format(opt.caption_model))

    // check compatibility if training is continued from previously saved model
    if vars(opt).get("start_from", None) is not None:
        // check if all necessary files exist 
        assert os.path.isdir(opt.start_from)," %s must be a a path" % opt.start_from
        assert os.path.isfile(os.path.join(opt.start_from,"infos_"+opt.id+".pkl")),"infos.pkl file does not exist in path %s"%opt.start_from
        model.load_state_dict(torch.load(os.path.join(opt.start_from, "model.pth")))

    return model

Instances