Hi, I am building a network containing a linear layer at the top of my architecture.
I notice that the number of input units need to be explicitly specified like:
self.fc1 = nn.Linear(5033, 30)
I know I can compute the number of output from my previous pooling layer. I was wondering if there is a more convenient way to define a network without computing number of units for each layer.
Thanks in advance