Ahmadzei's picture
added 3 more tables for large emb model
5fa1a76
For customizations that require changes in the training loop, you should
subclass [Trainer] and override the methods you need (see trainer for examples).