신경망의 기본 구현 및 학습 프로세스
데이터 로드 및 전처리
이미지 데이터는 지정된 폴더 구조에 따라 구성되어야 하며, 각 클래스별로 폴더를 분리하여 저장합니다.
data_dir = "./data"
transform = {
'train': transforms.Compose([
transforms.Resize((64, 64)),
transforms.ToTensor()
]),
'valid': transforms.Compose([
transforms.Resize((64, 64)),
tra ...
6월 27일 19:56에 게시됨
PyTorch 텐서 차원과 브로드캐스팅 완벽 이해
텐서의 차원 개념과 연산 후 차원 변화를 정확히 파악하는 것은 복잡한 신경망 구조를 이해하는 데 핵심입니다. 이 글에서는 PyTorch 텐서 생성과 브로드캐스팅 메커니을 심도 있게 다룹니다.
무작위 텐서 생성
딥러닝에서는 가중치 초기화나 모듈 통과 후 출력 형태 확인 등을 위해 무작위 텐서를 자주 생성합니다. 실제 이미지를 불러올 필요 없이 원하는 형태의 텐서 ...
5월 20일 06:26에 게시됨