About nn.Linear(16*5*5,120)

3

thank you very much！！！

Image Classification with custom Dataset