Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Tips:
VAN does not have an embedding layer, thus the hidden_states will have a length equal to the number of stages.