In VAE code I am trying to run, I get the error below, where the mismatched tensors seem like the same size. The mismatch error is triggered from the code below that is trying to instantiate a Linear layer. Appreciate any pointers on how to debug
RuntimeError: size mismatch, m1: [94 x 4608], m2: [94 x 4608] at …/aten/src/TH/generic/THTensorMath.cpp:961
snippets from source code:
self.fc1 = nn.Linear(h_dim, z_dim)
mu = self.fc1(h)