If my above assumption is correct, this is sadly not currently supported by TorchLayer. (The exact issue is that QCollection is not supported). If this is something you’d be interested in having, feel free to open a feature request on our github page!