diff --git a/advanced_source/cpp_cuda_graphs.rst b/advanced_source/cpp_cuda_graphs.rst index 41ebc84e..615ec20a 100644 --- a/advanced_source/cpp_cuda_graphs.rst +++ b/advanced_source/cpp_cuda_graphs.rst @@ -52,7 +52,7 @@ LibTorch(C++ 프론트엔드)에서의 CUDA 그래프 사용법은 다음과 매 위의 예시에는 순전파, 역전파, 가중치 업데이트가 포함되어 있습니다. 이 튜토리얼에서는 전체 네트워크 그래프 캡처를 통해 모든 계산 단계에 CUDA 그래프를 적용합니다. - 하지만 그 전에 약간의 소스 코드 수정이 필요합니다. 우리가 해야 할 일은 주 훈련 루프에서 +하지만 그 전에 약간의 소스 코드 수정이 필요합니다. 우리가 해야 할 일은 주 훈련 루프에서 tensor를 재사용할 수 있도록 tensor를 미리 할당하는 것입니다. 다음은 구현 예시입니다.