Skip to content

Cuda error: CUDA error: device-side assert triggered #2

@Stefano-retinize

Description

@Stefano-retinize

Hey @rd20karim thsnks for your help do far, I am having this error in the code line: Line 78

It seems it has to do with the number of classes the model is predicting or something.

Maybe related with that is that I noticed that the vocab size inside vocabulary.vocab_size_unk is different than the KIT model input_dim. So, I guess somehow the vocabulary class is building a longer vocabulary than what the trained model have.

Is there any chance you can share the class vocabulary object maybe as a .pkl (for the models model_ton5mfwh and model_u8qatt2y)? Maybe this is faster, since I guess is going to be more difficult to spot what's the error when is building this vocabulary class.

Thanks for your help.!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions