Python基于Pytorch Transformer实现对iris鸢尾花的分类预测，分别使用CPU和GPU训练