반응형 [pytorch] 프로젝트 구조 원칙 1. 데이터 처리(Data Handling)데이터 로딩 및 전처리: 데이터를 로드하고 전처리하는 코드는 data/ 또는 datasets/ 디렉토리에 위치할 수 있습니다. 데이터셋 클래스나 데이터 로더 함수들이 포함될 수 있습니다.데이터 변환(Transforms): 이미지나 텍스트 등의 데이터에 대한 변환 코드는 transforms/ 디렉토리에 위치할 수 있습니다.2. 모델 정의(Model Definition)신경망 모델: 주로 models/ 디렉토리에 저장됩니다. 각 모델은 별도의 파일로 정의될 수 있습니다.모델 유틸리티 함수: 모델 관련 유틸리티 함수들은 필요에 따라 utils/ 디렉토리에 포함될 수 있습니다.3. 학습과 평가(Training and Evaluation)학습 코드: 주로 train.py.. 2024. 7. 3. 이전 1 다음 반응형