been having pytorch trouble for months