[Ultralytics] TensorRT 문서에서 더 이상 유효하지 않은 INT8 배치 2배 참조 제거

PR 링크: ultralytics/ultralytics#23911 상태: Merged | 변경: +2 / -2

들어가며

코드가 변경되었는데 문서가 이전 동작을 설명하고 있으면, 사용자에게 혼란을 줍니다. Ultralytics의 TensorRT INT8 export 문서는 "캘리브레이션 시 배치 크기가 자동으로 2배로 설정된다"고 안내하고 있었지만, 실제 코드에서 이 로직은 이미 제거된 상태였습니다. 이번 PR은 문서를 현재 코드 동작에 맞게 수정합니다.

핵심 코드 분석

캘리브레이션 배치 설명 수정

Before:

During calibration, twice the `batch` size provided will be used. Using small
batches can lead to inaccurate scaling during calibration. ... so the `batch`
size is doubled automatically. If no batch size is specified `batch=1`,
calibration will be run at `batch=1 * 2` to reduce calibration scaling errors.

After:

Using small batches can lead to inaccurate scaling during INT8 calibration.
This is because the process adjusts based on the data it sees. Small batches
might not capture the full range of values, leading to issues with the final
calibration. Using a larger batch size helps ensure more representative
calibration results.

배치 2배 자동 적용 설명을 제거하고, "작은 배치가 부정확한 캘리브레이션으로 이어질 수 있으니 큰 배치를 사용하라"는 일반적 가이드라인으로 교체했습니다.

코드 예제 주석 수정

Before:

2. Sets max batch size of 8 for exported model, which calibrates with
   `batch = 2 * 8` to avoid scaling errors during calibration.

After:

2. Sets max batch size of 8 for exported model and INT8 calibration.

왜 이게 좋은가

1. 문서-코드 정합성: 코드에서 제거된 동작을 문서에서도 제거하여 사용자의 혼란을 방지합니다. "문서에는 2배라고 되어 있는데 왜 실제로는..."과 같은 이슈를 사전에 차단합니다.

2. 실용적 가이드 유지: 배치 크기가 캘리브레이션 정확도에 영향을 미친다는 핵심 메시지는 유지하면서, 구체적인 자동화 메커니즘 대신 사용자가 직접 적절한 크기를 선택하도록 안내합니다.

정리

TensorRT INT8 캘리브레이션 문서에서 더 이상 존재하지 않는 "배치 2배 자동 적용" 설명을 제거하고, 현재 동작에 맞는 가이드라인으로 교체한 문서 정비 PR입니다.

참고 자료

• ultralytics/ultralytics#23911
• TensorRT Export 문서

---

이 글은 AI(Claude)의 도움을 받아 작성되었으며, 실제 PR의 코드 변경 사항을 기반으로 분석한 내용입니다.