분석 Python/Pytorch
[TIP / Pytorch] torch class name 얻는 방법
데이터분석뉴비
2020. 10. 31. 14:32
728x90
torch class 이름을 얻고자 할 때, 삽질을 하였기 떄문에, 블로깅
다른 사람들은 삽질을 덜 하셨으면 한다!
layer = torch.nn.Linear(10,5) layer.__class__.__name__
728x90