Hey I’m doing a project similar to this but for myself and I’m also running into this error.
More like this tho:
RuntimeError: The size of tensor a (128) must match the size of tensor b (4096) at non-singleton dimension 1
I can’t do large batch sizes because my computer is really weak.
I was wondering if you could help
