djiajunustc / TransVG

157 stars 26 forks source link

Cannot load checkpoints correctly #30

Closed zzh-tech closed 2 years ago

zzh-tech commented 2 years ago

When I run evaluation code, I cannot load the checkpoints correctly.

Missing key(s) in state_dict: "visumodel.transformer.encoder.layers.0.self_attn.in_proj_weight", "visumodel.transformer.encoder.layers.0.self_attn.in_proj_bias" ... Unexpected key(s) in state_dict: "transformer.encoder.layers.0.self_attn.in_proj_weight", "transformer.encoder.layers.0.self_attn.in_proj_bias"...

zzh-tech commented 2 years ago

It seems that the provided checkpoints are for training. Could you please release the checkpoints for evaluation? Thanks in advance!

zzh-tech commented 2 years ago

Thanks for sharing the checkpoints!

HelloWorld4747 commented 1 year ago

Hello, may I please ask you if you can share the checkpoints for evaluation with me by any chance? Thanks in advance! Looking forward to your response soon! Cheers!