티스토리 뷰
github.com/ultralytics/yolov3/issues/1098
Memory Friendly Mish() Activation · Issue #1098 · ultralytics/yolov3
Hello, I tried yolov4.cfg on 2080Ti with input size 448*448, the training runs out of cuda memory even with batch size = 8 Have you tried the largest batch size value which won't break the cuda...
github.com
Mish activiation function을 사용하면 메모리 사용량이 크게 증가하는 것으로 보인다. Single Gpu에서 학습 가능한 모델을 만듦에 있어 Mish의 사용은 지양해야겠다.
'Deep Learning > PyTorch' 카테고리의 다른 글
| PyTorch to Onnx to TensorRT 과정을 위해 참고할만한 링크들 (0) | 2021.05.30 |
|---|---|
| RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` (0) | 2021.05.23 |
| [PyTorch & LibTorch] OpenCV Image to Normalized Tensor C++ 코드 (0) | 2021.02.18 |
| [PyTorch & TRTorch] TRTorch 오차 발생하는 경우 (0) | 2021.02.18 |
| [PyTorch] conv layer default initialization method (0) | 2021.02.12 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- cosine
- C++ Deploy
- 백트래킹
- FairMOT
- MOT
- 인공지능을 위한 선형대수
- PyCharm
- 조합
- ㅂ
- 단축키
- 이분탐색
- 위상 정렬 알고리즘
- Lowest Common Ancestor
- 파이참
- 백준 11437
- 순열
- 백준
- 자료구조
- 백준 11053
- 백준 1766
- LCA
- 가장 긴 증가하는 부분 수열
- 문제집
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 1 | 2 | 3 | 4 | |||
| 5 | 6 | 7 | 8 | 9 | 10 | 11 |
| 12 | 13 | 14 | 15 | 16 | 17 | 18 |
| 19 | 20 | 21 | 22 | 23 | 24 | 25 |
| 26 | 27 | 28 | 29 | 30 | 31 |
글 보관함