Hi.
Let us say I have 10 layers in a neural network. If the input is of type A, I want to execute all 10 layers. If the input if of type B, I want to execute only 8 layers and take the output. Is this easily implementable in pytorch? I tried this in tensorflow and caffe and faced some issues.
Thank you
Athindran