@override List<Value> parameters() { return [ ...bboxRegressionHead.parameters(), ...classPredictionHead.parameters(), ...faceEmbeddingHead.parameters(), // Add new head's parameters ]; }