- 博客(2)
- 收藏
- 关注
RSS订阅原创 初次使用单机多卡分布式训练遇到的问题汇总
* 注：可能model = ModelBuilder().cuda()也需要改成device = torch.device(“cuda:{}”.format(int(local_rank)));model = ModelBuilder().to(device)，后来有出现一些错误，我又乱改动了，最后可以运行的版本是用device = torch.device(“cuda:{}”.format(int(local_rank)));
2023-06-30 11:22:47
3083
原创 TransT环境配置问题
将from torch._six import string_classes, int_classes注释掉，然后添加int_classes = int、string_classes = str，查阅得知原因可能是torch版本和torchvision版本不对应，如果是版本不对应就要换版本了。按照README.md文件依次进行配置（包括激活命名为transt的conda环境），配置完成之后，在工程主目录TransT下面创建一个networks文件夹，然后将下载的预训练模型transt.pth移动到里面。
2023-06-22 20:48:54
315
2
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人