FC2カウンター FPGAの部屋 DeepLearningのモデル軽量化 Distiller を試す1
FC2ブログ

FPGAやCPLDの話題やFPGA用のツールの話題などです。 マニアックです。 日記も書きます。

FPGAの部屋

FPGAの部屋の有用と思われるコンテンツのまとめサイトを作りました。Xilinx ISEの初心者の方には、FPGAリテラシーおよびチュートリアルのページをお勧めいたします。

DeepLearningのモデル軽量化 Distiller を試す1

tkato さんの tkato's blog の”DistillerでDeepLearningのモデルを軽量化: Gradual Pruning編”を参考にして、DeepLearningのモデル軽量化 Distiller を試してみようと思う。

まずは、Docker で環境を構築する。
dockerhub のpytorch/pytorch の 0.4_cuda9_cudnn7 を使用する。

書いたDockerfile を示す。

FROM pytorch/pytorch:0.4_cuda9_cudnn7

RUN apt-get update && apt-get install -y vim


build.sh を示す。

docker build -t pytorch0.4cuda9 .


boot.sh を示す。

docker run -ti --rm --name distiller -p 6006:6006 -v /home/masaaki/Docker/distiller/workspace/:/workspace/ pytorch0.4cuda9 /bin/bash


Docker ディレクトリの下に distiller ディレクトリを作成し ./build.sh を起動した。
distiller_1_190219.png
distiller_2_190219.png

.boot.sh を実行して、Docker を起動した。
DistillerでDeepLearningのモデルを軽量化: Gradual Pruning編”を参考にして、Distiller を git clone する。
git clone https://github.com/NervanaSystems/distiller.git
distiller_4_190219.png
  1. 2019年02月20日 05:35 |
  2. DNN
  3. | トラックバック:0
  4. | コメント:0

コメント

コメントの投稿


管理者にだけ表示を許可する

トラックバック URL
http://marsee101.blog.fc2.com/tb.php/4449-4f4b084d
この記事にトラックバックする(FC2ブログユーザー)