您好，我复现您的代码，发现代码在运行过程中，加了预训练模型权重，但是还是会有很多报错 比如 提示mask错，更改后 还会提示 俩两个 张量不在一个设备，同时 代码不可以在 命令行直接设置 --train和 val 的位置。数据集用的 WUH
您好，我复现您的代码，发现代码在运行过程中，加了预训练模型权重，但是还是会有很多报错 比如 提示mask错，更改后 还会提示 俩两个 张量不在一个设备，同时 代码不可以在 命令行直接设置 --train和 val 的位置。数据集用的 WUH