티스토리 뷰
Deep Learning/PyTorch
[PyTorch] timm(rwightman/pytorch-image-models) 백본 스테이지별 채널 수 확인 코드
developer0hye 2021. 6. 18. 07:58import timm
model_name = 'your_model'
model = timm.create_model(model_name, pretrained=True, features_only=True).cuda()
model.feature_info.channels()
model.feature_info.channels() 값 확인해보면 스테이지별 채널 수를 확인할 수 있음.
'Deep Learning > PyTorch' 카테고리의 다른 글
| [PyTorch] AMP Loss dtype Issue (0) | 2021.06.23 |
|---|---|
| [PyTorch] DistributedDataParallel 예시 코드 및 참고 자료 모음 (4) | 2021.06.18 |
| [PyTorch] Gradient accumulation 예제 코드 (0) | 2021.06.16 |
| PyTorch to Onnx to TensorRT 과정을 위해 참고할만한 링크들 (0) | 2021.05.30 |
| RuntimeError: CUDA error: CUBLAS_STATUS_ALLOC_FAILED when calling `cublasCreate(handle)` (0) | 2021.05.23 |
댓글
공지사항
최근에 올라온 글
최근에 달린 댓글
- Total
- Today
- Yesterday
링크
TAG
- 백준 11053
- C++ Deploy
- 백트래킹
- ㅂ
- 이분탐색
- 순열
- FairMOT
- MOT
- 문제집
- 인공지능을 위한 선형대수
- 단축키
- Lowest Common Ancestor
- 백준 1766
- 자료구조
- LCA
- 파이참
- PyCharm
- 위상 정렬 알고리즘
- cosine
- 백준
- 백준 11437
- 가장 긴 증가하는 부분 수열
- 조합
| 일 | 월 | 화 | 수 | 목 | 금 | 토 |
|---|---|---|---|---|---|---|
| 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 |
글 보관함