Missing weights and trained model bin file is missing

#3
by Jay06 - opened

i am currently fine tuning the whisper-kannada-small model on custom dataset in Google Colab notebook. i am encountering two issues when loading a trained model: the first involves missing keys ('proj_out.weight') in the PyTorch model, indicating a potential problem during training where certain parameters are not being saved in the checkpoint. The second issue pertains to a binary format problem, where the model files ('pytorch_model.bin', 'tf_model.h5', etc.) are not being saved in the Output directory after training, as this .bin file is needed for evaluation process.
WhatsApp Image 2024-01-01 at 22.01.55.jpeg

WhatsApp Image 2024-01-01 at 22.04.36.jpeg

Sign up or log in to comment