Support for multiple input groups
Store initialiser networks and tasks in ModuleLists
so we can have a variable number of both. At the moment the different initialised groups must have the same embedding size and are simply concated into a longer list of inputs which is fed to the gnn.
A couple of things are left for the future:
- Support for both multiple inputs and aux tasks (will probably break atm, requires an umami update anyway)
- Support for concating the global representation to the per track reps for the aux tasks
Edited by Samuel Van Stroud