I am working on a Multi-Task text classifier, I want to make it config based i.e. I have an x input but not fix set of tasks. tasks are decided based on the config.
So how can I initialize different Classifiers based on config?
eg:- if my config has 2 tasks then 2 linear layers are initialized in model init and if my config has 5 tasks then 5 linear layers are initialized.