NeverDelay

1、搭建AlexNet網絡

2、如何使用自己的數據集使用網絡

——————————————————————————————————————————————————————

1、搭建AlexNet網絡

     網絡結構：

 定義網絡類：

• 定義初始化函數：定義網絡在正向傳播中所需要使用的層結構。

features:神經網絡層

使用nn.Sequential模塊：網絡層次比較多可以使用nn.Sequential簡潔代碼

kernel_size:卷積核個數，stride:卷積核步長，padding：邊緣補0的個數，如果是padding=[1,2]則為上下方各補一行0，左右兩側各補兩列零

classifier：全連接層

　　使用nn.Sequential模塊

　　使用nn.Dropout

　　初始化判斷init_weights（nn.Sequential中的參數）

• 定義前向傳播

使用torch.flatten(x,start_dim=1):展平處理，索引從1開始

• 定義初始化判斷函數