Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
Also, define a collate_fn, which will be used to batch examples together.