Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
It is therefore constrained by
the dimensions of the dummy input, and will not work for any other sequence length or
batch size.