pyTorch3 [에러][Pytorch] torch.cuda.OutOfMemoryError: CUDA out of memory 에러:torch.cuda.OutOfMemoryError: CUDA out of memory. Tried to allocate 50.00 MiB (GPU 0; 23.68 GiB total capacity; 22.27 GiB already allocated; 40.88 MiB free; 22.63 GiB reservedin total by PyTorch) If reserved memory is >> allocated memory try setting max_split_size_mb to avoid fragmentation. See documentation for Memory Management and PYTORCH_CUDA_ALLOC_CONF 원인:나는 현재 NVIDIA GeForce RTX 3090을 사용.. 2024. 7. 21. [에러][Pytorch] RuntimeError: Error(s) in loading state_dict 에러:모델 test 과정에서 ./pretrained 폴더의 .pth 파일을 torch.load 하는 부분에서 아래의 오류가 발생했다.model.load_state_dict(checkpoint['model_state_dict']) File "/~/DDAMFN/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1671, in load_state_dict raise RuntimeError('Error(s) in loading state_dict for {}:\n\t{}'.format(RuntimeError: Error(s) in loading state_dict for DDAMNet: Missing key(s) in state_.. 2024. 7. 21. [논문정리][GAN] Conditional GAN: Conditional Generative Adversarial Nets *본 내용은 논문의 상세한 분석이 아닌, 간단한 복기용 정리입니다. GAN Architecture 우선 일반 GAN부터 보자면, 목적함수는 다음과 같습니다. 왼쪽 판별자 박스 먼저 보겠습니다. P_data는 원본 데이터의 분포를 의미하는데요. 이 원본 데이터 중에서 한 개의 데이터인 x를 뽑아, D라고 나와있는 판별자에 이 하나의 이미지인 x를 넣습니다. 그럼 Discriminator가 x에 대해 판별을 합니다. 판별자는 진짜 이미지에는 1, 가짜 이미지에는 0을 부여하는데요. 그래서 판별자의 아웃풋은 0에서 1 사이의 확률값으로 나오고, 그것에 log를 취한 것의 기댓값을 구합니다. 오른쪽 생성자에서는 p_z에서 하나의 노이즈 데이터인 z를 뽑습니다. 이 p_z는 정규분포나 Uniform 분포를 주로 .. 2023. 11. 14. 이전 1 다음
