[TIP / Pytorch] torch class name 얻는 방법
2020. 10. 31. 14:32ㆍ분석 Python/Pytorch
728x90
torch class 이름을 얻고자 할 때, 삽질을 하였기 떄문에, 블로깅
다른 사람들은 삽질을 덜 하셨으면 한다!
layer = torch.nn.Linear(10,5) layer.__class__.__name__
728x90
'분석 Python > Pytorch' 카테고리의 다른 글
| [Pytorch] torch 유용한 함수 정리하기 (0) | 2020.11.30 |
|---|---|
| PyTorch Lighting + Ray tune (0) | 2020.11.07 |
| [TIP / Pytorch] calculate convolution output shae (conv2d , pooling) (Conv 아웃풋 값 (0) | 2020.10.31 |
| [TIP / Pytorch 1.5~] jit script, save, load (0) | 2020.10.29 |
| [TIP / Installation] requirements.txt 로 pytorch package 설치하는 방법 (1) | 2020.10.25 |