- 博客(2)
- 收藏
- 关注
RSS订阅原创 PyTorch查看pth.tar文件内容并用于pretrain
path中为文件路径import torchpath='LightGCN-PyTorch-master/code/checkpoints/lgn-lastfm-3-64.pth.tar'pretrained_dict = torch.load(path)print(pretrained_dict.keys())print(pretrained_dict['embedding_user.weight'])
2022-04-07 17:21:27
2118
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人