self .fc1_bn = nn.BatchNorm1d( 128 ). self .fc2 = nn.Linear( 128 , 10 ). def forward( self ,x):. x = self .conv1(x). x = F.relu( self .conv1_bn ...
確定! 回上一頁