Option to only return the last hidden layer output from models #342
Description
In many applications we only need the last layer and letting go of references to intermediate layers can save some memory during inference.
{{ message }}
In many applications we only need the last layer and letting go of references to intermediate layers can save some memory during inference.