finished epoch training/testing, bug still present

still converging to only one classification, still not sure why.