[ ] This template is only for bug reports, usage problems go with 'Help Wanted'.
[ ] I have thoroughly reviewed the project documentation but couldn't find information to solve my problem.
[ ] I have searched for existing issues, including closed ones, and couldn't find a solution.
[X] I confirm that I am using English to submit this report in order to facilitate communication.
Environment Details
pytorch
Steps to Reproduce
prepare dataset (csv custom)
python train.py >>> size mismatch for ema_model.transformer.text_embed.text_embed.weight: copying a param with shape torch.Size([2546, 512]) from checkpoint, the shape in current model is torch.Size([94, 512]).
✔️ Expected Behavior
training must be started
❌ Actual Behavior
why size mismatch for ema_model.transformer.text_embed.text_embed.weight: copying a param with shape torch.Size([2546, 512]) from checkpoint, the shape in current model is torch.Size([94, 512]).
Checks
Environment Details
pytorch
Steps to Reproduce
✔️ Expected Behavior
training must be started
❌ Actual Behavior
why size mismatch for ema_model.transformer.text_embed.text_embed.weight: copying a param with shape torch.Size([2546, 512]) from checkpoint, the shape in current model is torch.Size([94, 512]).