For some reason, this method doesn’t seem to work for me. I receive the following error:

RuntimeError: size mismatch, m1: [3584 x 7], m2: [25088 x 4096] at /pytorch/torch/lib/TH/generic/THTensorMath.c:1293

Did you face a similar issue? I don’t know why this should happen because after all, it’s using all the layers in a sequential fashion.