Splitting a CNN model to run in two devices
What model are you splitting?
In PyTorch it is possible to do so. Check the children function of a model in pytorch and reconstruct the layers for two sub models.
In a part of my research work, I am in need to split a deep Learning model(CNN) into two in order to run the model in two different devices in a distributed way. Here, I split a CNN model into two, A and B where the output of A will be the input for B. If anyone has an idea, please.