hcwei

About the parameters of the imported model and the current model are inconsistent

1

When I run train.py, This problem has arisen： ``` RuntimeError: Error(s) in loading state_dict for AoAModel: size mismatch for embed.0.weight: copying a param with shape torch.Size([10370, 1024]) from checkpoint, the...

Thank you for your excellent work! Do you have any plans to release the checkpoints for Stage 1 and Stage 2?