Open goodboyanush opened 4 years ago
@goodboyanush hi, sir, do you solve the problem?
@Raukk do we have solution for this?
I'm having the same problem on tensorflow 2.2.
I'm sorry, between work and life, I have not had the time to maintain this project.
It's fine, it is also a hard project to maintain as the tensorflow API changes a lot. I found out this problem is algo present in the original kerassurgeon package (see this issue ).
There, the person bringing up the issue managed to overcome it by putting two simultaenous jobs on the two branches that merge together. This, however, requires the user to identify which layers are added beforehand and prune them together, which should be automated to some point.
I think this could probably be achieved by changing the Surgeon class when the graph is recursively rebuilt, but I have yet to become more familiar with the library.
I tried the
ResNet18
model fromkeras-applications
(tf.keras.applications
) and I get an error in theAdd
layeryes, I am trying to remove three channels from the first convolution layer