Network in the network NiN hands-on deep learning v2 pytorch

1. Network NiN in Network

The network NiN in the network is based on the attempted improvement of AlexNet, and the fully connected layer is removed.
The network NiN in the network is not used much later, but the idea of ​​removing the fully connected layer is adopted.
insert image description here
The biggest problem with the fully connected layer is that

  1. uses a lot of memory
  2. Takes up a lot of computing bandwidth
  3. Convergence is fast, but can lead to overfitting
    insert image description here
    insert image description here
    insert image description here
    insert image description here
    insert image description here

2. Code implementation

insert image description here

insert image description here
insert image description here
insert image description here

3. Q&A

    1. nvidia-smiIt is nvidia's system management interface, where smi is the abbreviation of System management interface, which can collect various levels of information and view video memory usage.
    1. AlexNet, NiN still need Softmax and need to be classified.
    1. Why does NiN need two 1x1 convolutional layers, the high probability is the result of training parameters.

reference

https://www.bilibili.com/video/BV1Uv411G71b?p=1

Guess you like

Origin blog.csdn.net/zgpeace/article/details/124139383