cannot load kernels from composite in IIPR

[AlessandroLovo]

The problem is that propabilistic_regression adds the extra sigma activation layer by creating a nested sequential model.
Then when committor_projection_NN tries to do model.layers[0] the whole thing breaks