Refactor model export part 2: Add a dedicated forward function for mo… · openvinotoolkit/training_extensions@2e6d225