1. 학습
    2. /
  2. 강의
    3. /
  3. PyTorch로 AI 모델 효율적으로 학습시키기
Connected

연습 문제

분산 학습을 위한 데이터셋 준비

정밀 농업 시스템을 위한 데이터셋을 전처리했습니다. 이 시스템은 농부들이 작물 상태를 모니터링하는 데 도움을 줍니다. 이제 DataLoader를 생성하여 데이터를 로드하고, GPU가 사용 가능한 경우 분산 학습을 위해 데이터를 GPU에 배치해 보겠습니다. 이 연습 문제는 실제로 CPU를 사용하지만, CPU와 GPU 모두 동일한 코드를 사용합니다.

다음 데이터가 사전에 로드되어 있습니다:

  • 농업 이미지가 포함된 샘플 dataset
  • accelerate 라이브러리의 Accelerator 클래스
  • DataLoader 클래스

지침

100 XP
  • 사전에 정의된 dataset에 대한 dataloader를 생성하세요.
  • accelerator 객체를 사용하여 dataloader를 사용 가능한 디바이스에 배치하세요.