Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Some weights of RobertaModel were not initialized #18

Open
timbmg opened this issue May 9, 2024 · 1 comment
Open

Some weights of RobertaModel were not initialized #18

timbmg opened this issue May 9, 2024 · 1 comment

Comments

@timbmg
Copy link

timbmg commented May 9, 2024

I have downloaded the AlignScore-base.ckpt and am initialzing AlignScore like so:

align_evaluator = AlignScore(
        model="roberta-base",
        batch_size=4,
        device="cpu",
        ckpt_path="/path/to/AlignScore-base.ckpt",
        evaluation_mode="nli",
)

However, this gives me the following warning, which concerns me a bit:

Some weights of RobertaModel were not initialized from the model checkpoint at roberta-base and are newly initialized: ['roberta.pooler.dense.bias', 'roberta.pooler.dense.weight']
You should probably TRAIN this model on a down-stream task to be able to use it for predictions and inference.

Is this expected? Will the evaluation be correct? I am assuming that these are not required because different heads are used, but woudl like to make sure.

@maxyousif15
Copy link

I've had a similar warning, but I'm assuming it is safe to ignore from this issue, which seems identical, despite a different warning message?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants