Also, I just read that Image folder creates it’s own labeling so I ran
data['train'].class_to_idx

and got the following result which I think is the problem.
{'.ipynb_checkpoints': 0, '1': 1}

can I change the classes after the fact like I tried above?