AssertionError: X and Y should be the same shape [CPU Mode]
Shelomoh1 opened this issue · comments
@justchenhao Can you please help me out. When I run the demo.py it runs and produces the output but when I run the script train.py to train with the base method, I get the error "AssertionError: X and Y should be the same shape" during the 400th iteration of the first epoch. The training fails at that point. I have attached an image of the error.
This problem has to do with the version of visdom. I downgraded visdom and the error disappeared.