- 博客(4)
- 收藏
- 关注
RSS订阅原创 【pytorch的 unsqueeze和view函数】
记录pytorch的shape是怎么变化的，主要是unsquezze函数代码如下：a = torch.tensor([[[1,2,3,4],[0,0,0,0],[7,8,9,10]],[[1,4,9,16],[16,25,36,49],[49,64,81,100]]],dtype=torch.float)len_q = a.size(1)padding_mask = a.eq(0)print('1,',padding_mask)print('2,',padding_mask[:,:,0])pr
2022-04-20 09:10:04
905
原创 【深度学习训练使用tensorboard】
pycharm+服务器+pytorch+tensorboard：查看train test loss：主要包括pycharm里的代码部分和ssh里的操作代码部分提示：这里描述项目中遇到的问题：例如：数据传输过程中数据不时出现丢失的情况，偶尔会丢失一部分数据APP 中接收数据代码：from torch.utils.tensorboard import SummaryWriterwriter = SummaryWriter('/data/xxxx/xxxx/xxxx/xxxx/runs')#在
2022-03-22 23:26:11
2571
原创 Linux服务器上添加了清华镜像也无法安装pytorch解决方案
Linux服务器上添加了清华镜像也无法安装pytorch解决方案创建虚拟环境到安装包建立虚拟环境创建虚拟环境到安装包今天遇到加镜像也还是安装不了pytorch的问题，首次尝试手动下载并安装，仅此记录供大家参考。建立虚拟环境1.conda create -n [虚拟环境名称] python=3.82.conda activate [虚拟环境名称]3.（已经默认配置了清华镜像）conda install pytorch torchvision torchaudio cudatoolkit=11.0
2021-01-05 15:52:19
948
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人