(16*5*5, 120),
nn.ReLU(),
nn.Linear(120, 84),
nn.ReLU(),
... Linear(in_features=10, out_features=10, bias=True)
(3): Linear(in_features=10, out_features=10, ...bias=True)
(4): Linear(in_features=10, out_features=10, bias=True)
(5): Linear(in_features=10...(in_features=10, out_features=10, bias=True)
(8): Linear(in_features=10, out_features=10, bias=True...)
(9): Linear(in_features=10, out_features=10, bias=True)
(10): Linear(in_features=10, out_features