下面命令是查看是否使用GPU。

3.直接运行py文件，会直接将转换后的文件保存在image文件夹下。2.可以将之前的要穿换的images文件名改为oldimage。发现是直接将png图片扩展名修改成jpg，造成图片失真问题。解决： 使用OpenCV模块将图片读出再写入，具体代码如下。1.在数据集的目录下创建transform.py 文件。4.验证结果，看是否还有问题。训练自己数据集时出现。

运行结果如下，也可以在import torch后直接使用第四条命令。返回设备名称，从索引0开始。

记下我们所需要的版本我们去清华源网站去寻找我们的包，我们只需要找三个，pytorch，torchvison和torchaudio这三个包，我们直接输入后面的版本号在清华源里面寻找，Ctrl+f可以调出网页搜索，找到我们需要的包注意pytorch1.7.1中下面的python版本和cuda版本，不要去安装cpu的就ok，我们使用的是python3.8和CUDA10.1，框出来的就是我需要的版本。在p

记下我们所需要的版本我们去清华源网站去寻找我们的包，我们只需要找三个，pytorch，torchvison和torchaudio这三个包，我们直接输入后面的版本号在清华源里面寻找，Ctrl+f可以调出网页搜索，找到我们需要的包注意pytorch1.7.1中下面的python版本和cuda版本，不要去安装cpu的就ok，我们使用的是python3.8和CUDA10.1，框出来的就是我需要的版本。在p

这里先描述一下任务，这里是昨天晚上跑了一个200 epochs的训练，这里是没有添加任何模块实验结果：这里是训练了167轮之后没有什么提升就给停止了，这里说最好的模型是第67轮的模型，这里去看下，我们每次跑实验都要记录一下哪些参数，应该有哪些参数的设置，然后就是加模块跑下实验，看最后的评价标是否有提升。

下面命令是查看是否使用GPU。

在目录下新建文件，内容如下：SEAttention.py 代码如下在中，导入模块在tasks.py头部粘贴代码。