InferenceRecorder improvements