Hi, I have a small question, I want to exit my network, but not it the end, let’s say I’m certain in some level that my result is correct after 3 layers, and I don’t want to do the other 2 layers, How can I do it?
I have the condition, and I’m counting the number of time it should exit, But I’m puzzled how to exactly exit the graph, let’s say I have batch of 32 and 10 of them are good enough after 3 layers, how can I stop their processing (in pytorch of course)?
Just to make things clear, I’m not asking about the theory, I’m asking about how to finish processing and adding to the next data in line in pytorch.
Thanks.