[Bug]: 训练出现的问题

[zzy914zzy]

Bug

您好！我按照readme里面的流程配置了环境，修改了数据及结构后运行CUDA_VISIBLE_DEVICES=0 accelerate launch main.py训练出现问题，请问怎么解决啊

环境信息

No response

补充信息

No response

[zzy914zzy]

询问GPT的回答是torch._six已经过时，需要更新相关库或修改代码以兼容 PyTorch 最新版本。我目前是按照readme中的版本要求配置的pytorch==1.12.1 torchvision==0.13.1

[zzy914zzy]

已解决