search

Modalities / modalities

A framework for training multimodal foundation models.
MIT License
57 stars 5 forks source link

Transform Trainer into a component #83

Open le1nux opened 5 months ago

le1nux commented 5 months ago

Currently, the Trainer class is instantiated within the main.py. This forces us to feed in all the dependent components such as model, optimiser, scheduler into the trainer object within main.py during instantiation.

A better way would be to instantiate the trainer object as a component that is part of the hierarchical instantiation.