How to training again GOT-OCR2.0 in Stage 2, Stage 3

[C0NGTRI123]

Hello, I want to ask again to confirm. After I training again in stage 1 in Vary_tiny, they return me the weight .pth. The weight from Huggingface you provide, it may be the weight from pretrain. Because I read your paper after stage 1, you get vision encoder continue training with Qwen-0.5B in stage 2 and remove old LLM(OPT-125m) in stage 1. But I don't see this code. Can you explain me that

[Ucas-HaoranWei]

You should extract the encoder weights in stage-1 .pth, and load it to the next stage models.

[han508]

Hi, how to extract stage-1.pth in model.safetensor @Ucas-HaoranWei