generator
[DL] RuntimeError: Expected a 'cuda' device type for generator but found 'cpu'
File "/-/.conda/envs/torch38/lib/python3.8/site packages/torch/utils/data/dataloader.py", line 444, in __iter__ return self._get_iterator() 이 부분 부터 에러가 시작됐다. self.dataloader = DataLoader( dataset=self.dataset, batch_size=self.batchsize, shuffle=True, num_workers=self.num_workers, # 이 부분 추가 generator=torch.Generator(device='cuda') ) generator=torch.Generator(device='cuda') 위의 코드를 추가하여 수동으로 Genera..
[GAN] 생성 모델 GAN의 이해
GAN (Generative Adversarial Network) 이미지 생성에 자주 쓰이는 모델로 학습 데이터 셋과 유사한 이미지를 만들도록 학습됨. unlabeled data를 사용하며 Generator는 데이터의 확률 분포를 학습함 파란색 점선: Discriminator distribution 검은색 점선: Real data distribution 초록색 실선: Fake data distribution Generator가 실제 이미지의 분포를 학습하는 과정이며 Discriminator는 Generator의 학습 방향을 안내함 백쿼리 backquery 기본적으로 신경망은 데이터셋의 정보를 감소하고 정제하고 축약하는 과정에 쓰인다 -> CNN의 classification을 예시로 들면 3x32x32의 ..
