에러2 [에러][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. 이전 1 다음