Thank you. that’s what i was confuse about exactly.
I want to gradient flow back to conv1 but i noticed that when i also compute the gradient inside the costumefunction it just makes the process super slow, so that is why i was wondering maybe i kill it inside the costumefunction.
Since i need the gradient to flow in the net, i geuss i have no choice to compute it inside the costumefunction as well…