running_mean should contain 1 elements not 256
For context, I’m using this tutorial for Image Captioning GitHub and I’m now trying to fine tune it, but to my dataset.
For context, I’m using this tutorial for Image Captioning GitHub and I’m now trying to fine tune it, but to my dataset.