SVD 코드 작성 중, pytorch로 구현시 한 epoch당 2분 가까이 걸리는데 np로된 dot 연산으로 구현하면 1회당 8초가 걸리는 현상을 발견했다,,
GPU를 쓸 때는 pytorch
np.dot 연산은 numpy
Es muss sein!
0개의 댓글
관련 채용 정보
SVD 코드 작성 중, pytorch로 구현시 한 epoch당 2분 가까이 걸리는데 np로된 dot 연산으로 구현하면 1회당 8초가 걸리는 현상을 발견했다,,
GPU를 쓸 때는 pytorch
np.dot 연산은 numpy