반응형
체크해야할 부분
1. 배치 사이즈 문제
2. Num_worker 늘리기. Num_worker는 data loader가 데이터를 읽어들일때 사용할 쓰레드의 갯수입니다. 보통 (사용하는 gpu 개수)*4로 해놓고 +- 조절해서 쓴다고 한다.
3. Pin_memory=True
Dataset을 CPU 텐서로 loading 하고 있다면 pin_memory = True를 사용해서 device transfer을 빠르게 해줄 수 있다.
4. 데이터가 SSD에 있는지?
lsblk -d -o name,rota
df -h 를 사용해서 데이터가 있는 위치가 SDD인지 확인한다.
5. Data normalizing
0~255 이미지보다 (0,1) normalize 하는게 더 빠르다 한다.
+ 해당 변수가 어디서 처리되고 있는지 확인한 후 gpu로 옮겨줄 수 있다면 옮겨주기
+ visualization 최소화하기. ( (ex) 텐서보드 이미지 저장하는거)
반응형
'컴퓨터공학' 카테고리의 다른 글
| 삼성 SW 코테 기출, 어항정리 (0) | 2022.03.22 |
|---|---|
| 카메라 stabilization 알고리즘 기초 (1) | 2022.02.16 |
| Pytorch loss nan 문제 (0) | 2022.02.11 |
| Pytorch 시드 고정하는 법 (0) | 2022.02.10 |
| 쌩판 노베이스 비전공자를 위한 딥러닝 공부 순서 (23) | 2022.02.09 |
