Hello, have you figured this out?
I am dealing with the exact same problem where I want to modify hidden layers of a pretrained network s.t. they output auxiliary loss values.
Any help will be very much appreciated