Now training works! Must get testing to work next.