your can try to run all of your hidden layers in your forward function. here’s the sudo code:
x = F.relu(self.conv1(x))
x = F.relu(self.conv2(x))
x = F.relu(self.conv2(x))
and then apply a for loop on the output of every layer and iterate it up to the number of layers you have and keep displaying all of he layers.