网站首页
编程语言
数据库
系统相关
其他分享
编程问答
transformer-model
2024-07-30
视觉变换器模型未按应有的方式进行训练
这是我使用Pytorch构建的视觉转换器的代码。该模型的交叉熵为2.31,准确度约为10%。这在所有时代都是一样的。因此,该模型无法训练。请让我知道我做错了什么,如果可能的话,请发送修改后的代码。预先感谢!PS:该模型是在MNIST上训练的#-*-coding:utf-8-*-"""Createdon