CNTK工具使用總結(jié)
一、下載及安裝 CNTK(Cognitive Toolkit)是微軟公司開發(fā)的深度學(xué)習(xí)工具箱,本文使用已編譯好的CNTK工具版本,如果想自己編譯可以的話,可以使用源碼版本。官網(wǎng)安裝指南:
https://github.com/Microsoft/CNTK/wiki/Automated-binary-install 。
二、使用指南 以MNIST數(shù)據(jù)集為例,簡單介紹下CNTK的使用。
1. 打開Win10系統(tǒng)下的命令提示符窗口,輸入C:\local\cntk\Scripts\cntkpy34.bat,運(yùn)行批處理文件cntkpy34.bat文件,打開CNTK-Python環(huán)境。
2. 在CNTK-Python環(huán)境下,切換C:\repos\CNTK\Examples\Image\DataSets\MNIST文件夾為當(dāng)前目錄,
輸入python install_mnist.py命令,運(yùn)行Python腳本install_mnist.py下載MNIST訓(xùn)練集和測試集,并轉(zhuǎn)換為.txt文檔。
3. 將當(dāng)前目錄切換到C:\repos\CNTK\Examples\Image\GettingStarted,輸入命令
cntk configFile=01_OneHidden.cntk運(yùn)行配置文件。配置文件由微軟的BrainScript語言編寫,包含訓(xùn)練過程模塊(trainNetwork)和測試過程模塊(testNetwork),trainNetwork模塊里包含對網(wǎng)絡(luò)結(jié)構(gòu)的定義。運(yùn)行該配置文件即可用訓(xùn)練集訓(xùn)練網(wǎng)絡(luò)模型,再用測試集測試之。最終訓(xùn)練完的模型和中間的訓(xùn)練結(jié)果存儲在 .\Output\Models 文件夾下。當(dāng)模型訓(xùn)練完后,可以在C++或者C#語言中(其他語言也可,待探索)部署訓(xùn)練好的模型,在其他的數(shù)據(jù)集中得到預(yù)測結(jié)果。
4. 3中的訓(xùn)練過程和測試過程也可以用Python語言編寫的.py文件實(shí)現(xiàn),將當(dāng)前目錄切換到C:\repos\CNTK\bindings\python\examples\MNIST,輸入命令
python SimpleMNIST.py運(yùn)行.py文件,進(jìn)行網(wǎng)絡(luò)模型的訓(xùn)練與測試。這種方式下模型訓(xùn)練完后,利用何種語言調(diào)用該模型進(jìn)行部署還有待進(jìn)一步探索。
本站僅提供存儲服務(wù),所有內(nèi)容均由用戶發(fā)布,如發(fā)現(xiàn)有害或侵權(quán)內(nèi)容,請
點(diǎn)擊舉報。