원본이라고 생각되는 GitHub - hanyoseob/youtube-cnn-002-pytorch-unet: [CNN PROGRAMMING] 002 - UNET 를 돌려보니
제 머신에서 batch 4에서 약 6G, batch 1에서 4.3G 정도 사용되네요.
올려주신 코드도 4.3G 정도네요. CUDA 버전(11.0)이나 파이토치 (1.7)버전 디바이스 정보(V100), 다른 프로세스가 돌고 있지는 않았는지 도 확인해보셔야 할 것 같습니다.
1개의 좋아요