Pytorch Audio Classification Using Machine