사담

[python bug fix diary] model.fit could not create a view primitive descriptor

다진마늘a 2019. 9. 18. 12:25

 

 

 

겪고나니 다른사람은 안그랬음 좋겠어서 올림. ...

 

CBAM(ㅋㅋ 한국인이 만든 모델이라 이름이 시밤임ㅋㅋ)을 적용해보려고 만져본 것까진 좋았는데 

 

message: could not create a concat primitive descriptor, in file tensorflow/core/kernels/mkl_concat_op.cc

 

라며 에러가 난다.

 

이건 뭐 소스코드를 만져서 해결해야하는 게 아니라 GPU 할당문제라고 해서 나는 tensorflow를 다시 깔았다 설치했다가 고친 소스 넣었다 뺐다 했는데... github 페이지 젤 아래에 있는 방식이 해결해주더라

 

나는 conda+ pycharm+ python을 쓰고있어서

conda prompt를 열어서

 

conda update -n base -c defaults conda

conda update mkl

 

이거 두 개 업데이트 해주고 나니까 바로 되더라.

 

그리고 그렇게 고치고 사용해본 CBAM모델은 내가 쓰는 데이터엔 잘 맞지 않았는지 0.02 acc를 보임 시무룩.

 

 

https://github.com/tensorflow/tensorflow/issues/17494

 

TF1.6 MKL bug in concatenation op · Issue #17494 · tensorflow/tensorflow

System information Have I written custom code (as opposed to using a stock example script provided in TensorFlow): Yes. OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Linux CentOS 7 Tenso...

github.com