定义了模型架构需要的参数和激活函数:
// AlexNet define
#ifndef __SLQ_NET_MACRO_H__
#define __SLQ_NET_MACRO_H__
#include <cmath>
namespace slqDL {
namespace slqAlexNet{
#define EpochLoop (100)
#define AccuracyRate (0.56)
#define Alpha (0.01f)
#define LoopError (0.001f)
#define EspCNN (1e-8f)
#define poolSpace (3)
#define poolStride (2)
#define TrainImgNum (40000)
#define TestImgNum (10000)
#define inMapHigh (227)
#define inMapWidth (227)
#define inMapNum (3)
#define inMapSize (51529) // 227*227
#define inUnitNum (154587) // 227*227*3
#define c1MapHigh (55)
#define c1MapWidth (55)
#define c1MapNum (96)
#define c1MapSize (3025) // 55*55
#define c1UnitNum (290400) // 55*55*96
#define c1ConvHigh (11)
#define c1ConvWidth (11)
#define c1ConvDeep (3)
#define c1ConvStride (4)
#define c1ConvTensor (121)