Now getting this:
RuntimeError: Expected 4-dimensional input for 4-dimensional weight [1, 1, 69, 69], but got 5-dimensional input of size [10001, 1, 1, 69, 69] instead
How do I take 1 dimension down from the input? how to I remove the 10001 from the input?
Thanks!