DeepNet的基本框架包含三个部分,也就是它里头的三个文件夹:deepnet,cudamat和eigenmat。
其中deepnet中包括了DeepNet的接口函数,cudamat和eigenmat分别包含了DeepNet的库函数。
deepnet
deepnet文件夹中包含的是用Python实现的接口函数。主要有:
trainer.py:训练网络时使用的就是这个接口函数,训练的模型主要有四种:FEED_FORWARD_NET,DBM,DBN,SPARSE_CODER以及FAST_DROPOUT_NET,具体模型的类型在model.pbtxt中定义。使用方法:
python trainer.py model.pbtxt train.pbtxt eval.pbtxt
其中train.pbtxt和eval.pbtxt在上一章中提到过,在此不再赘述。
上