Запуск Alexnet за допомогою MxNet - python, mxnet

Я намагаюся грати з кодом алекснет в/ mxnet / example / image-classification / каталог каталогів з використанням MxNet Framework. Я не є експертом в AI. Чи можна пояснити, як запустити його за допомогою GPU? Я спробував наступне для одного GPU:

python alexnet.py --network resnet --num-layer 110 --пакет-розмір 128 - gpus 0

Це didn "t нічого. У мене є HPC фону. Я хочу, щоб перевірити масштабованість цієї рамки на вузол і через вузли (розподілені). Будь-яка допомога буде оцінена.

Дякую,

Відповіді:

alexnet.py (разом з іншими файлами Python у прикладах / папці класифікації зображень / символів) повертає тільки символи, які представляють мережу.

Спочатку завантажте та скасуйте архів з вашого набору даних:

/mxnet/example/image-classification/data# wget http://www.image-net.org/image/whatever-zip-or-tar-file
/mxnet/example/image-classification/data# unzip whatever-zip-or-tar-file

Перетворення формату даних у RecordIO:

/mxnet/example/image-classification/data# python ../../../tools/im2rec.py --list True --recursive True --train-ratio 0.95 mydata tiny-imagenet-200
/mxnet/example/image-classification/data# python ../../../tools/im2rec.py --num-thread 16 mydata tiny-imagenet-200

Використовуйте сценарій train_imagenet.py, щоб тренуватися на alexnet (ви можете переключитися на будь-який з інших символів, якщо хочете):

/mxnet/example/image-classification/data# cd ..
/mxnet/example/image-classification# python train_imagenet.py --network alexnet --data-train /mxnet/example/image-classification/data/mydata_train.rec --data-val /mxnet/example/image-classification/data/mydata_val.rec --num-layers 110 --batch-size 64 --gpus 0

Погляньте на README для більш детальної інформації.