pytorch: model의 device 얻는 방법

djlee·2023년 3월 17일

0

model에 적용된 device를 얻고 싶을 때는 아래 코드를 사용하면 돼요.

next(model.parameters()).device

이전 포스트

pytorch: NVIDIA RTX 4090 DDP freeze 현상 해결 방법

다음 포스트

pytorch: gpu or cpu device 설정하는 방법

0개의 댓글