분석 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