Skip to content

Support for multiple input groups

Samuel Van Stroud requested to merge svanstro/multiple-inputs into main

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

Merge request reports