Replacing a block of layers

[nscotto]

Hi,

I am writing a module for searching and replacing a block of layers.
I am struggling at the moment to find how to reconnect everything so I wanted to ask you whether if you could kindly give me some directions with it.

So basically I have tf.keras.Model, I have found the block to replace, I know it's old_block_layer_start, old_block_layer_end, and I have a new block, coming up as a tf.keras.Model for making things simpler, or maybe else as a tuple new_block_layer_start, new_block_layer_end.

I am trying to understand the mask part of your project and whether that's necessary for me or not.

Currently I have been able to replace the head of a model by a new block (only connection from model to new block input), but haven't been successfull in replacing the middle of a model (connection from model to new block input, then new block output to model), I am only replacing things like node.outbound_layers and node.inbound_layers.

Thank you for reading.