Deep Learning

여기서 사용하는 Flux는 v0.10.1 이다. Model-zoo의 MNIST 예제를 참고(https://github.com/FluxML/model-zoo/blob/master/vision/mnist/mlp.jl) zoo의 mnist예제 mlp가 gpu모드에서 재대로 동작하지 않는데 주요 원인은 onecold function이 gpu모드에서 문제를 일으킨다. 위 예제에서 accuracy function에 onecold를 사용하고 train!에서 accuracy를 사용하기 때문에 문제가 된다. 따라서 아래와 같이 바꿔 준다 accuracy(x,y) = mean(onecold(m(x)) .== onecold(y)) ==> accuracy(x,y) = mean(onecold(m(x)|>cpu) .== onecold(y|>cpu)) 또한 scalar연산을 할 때 […]

Flux가 CuArrays등을 사용하기 위해서는 관련된 패키지 설치시 dependency를 잘 맞춰 줘야 한다. 현재 설치된 Julia 1.2.0 기준으로 하며 아래와 같이 한다 CUDAdrv 3.1.0을 먼저 설치한다. 기존에 관련된 패키지가 있는 경우 버전 업/다운이 일어 난다.무조건 높은 버전이 좋지 않음을 알 수 있다. 나의 경우 높은 버전을 설치 했더니 아래 처럼 설치된 관련 패키지 버전이 다운 그레이드