RuntimeError: output with shape [16, 8, 1, 1] doesn't match the broadcast shape [16, 8, 1, 64]

#7
by Dharma610 - opened

trainer.evaluate() is throwing an error "RuntimeError: output with shape [16, 8, 1, 1] doesn't match the broadcast shape [16, 8, 1, 64]"

image.png

Sign up or log in to comment