[질문] with tf.device('/cpu:0'):

[zidane100e]

안녕하세요

간단한 방법으로 한글 단어 간의 유사도를 비교 할 수 있는 효과적인 방법을 알려 주시고, 좋은 코드를 이렇게 올려주셔서 감사합니다. 해당 모델을 적용해 보려고 테스트하다 궁금한 점이 생겨 질문 드립니다

train.py 에 있는 with tg.device('/cpu:0'): 부분이 어떤 역할을 하는지 설명해 주실 수 있으신지요? 단순하게 gpu를 쓰면 에러가 생기더 군요.

[dongjun-Lee]

안녕하세요. 메일 주셔서 감사합니다. tf.device는 특정 디바이스에서 연산이 실행되도록 설정하는 함수입니다. Tensorflow 몇몇 연산은 gpu에서 동작하지 않는데, 그 중 하나인 embedding_lookup을 사용하기 때문에 해당 부분을 gpu로 바꾸면 에러가 나는 것으로 보입니다.

감사합니다.

2019년 6월 28일 (금) 오후 5:36, zidane100e [email protected]님이 작성:

안녕하세요

간단한 방법으로 한글 단어 간의 유사도를 비교 할 수 있는 효과적인 방법을 알려 주시고, 좋은 코드를 이렇게 올려주셔서 감사합니다. 해당 모델을 적용해 보려고 테스트하다 궁금한 점이 생겨 질문 드립니다

train.py 에 있는 with tg.device('/cpu:0'): 부분이 어떤 역할을 하는지 설명해 주실 수 있으신지요? 단순하게 gpu를 쓰면 에러가 생기더 군요.

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub https://github.com/dongjun-Lee/kor2vec/issues/3?email_source=notifications&email_token=ABRV6CSPBNFAFMGRPLF2NO3P4XEQFA5CNFSM4H4DKOT2YY3PNVWWK3TUL52HS4DFUVEXG43VMWVGG33NNVSW45C7NFSM4G4IMGDQ, or mute the thread https://github.com/notifications/unsubscribe-auth/ABRV6CWV3OAAPXKMFOXJSHDP4XEQFANCNFSM4H4DKOTQ .