- 博客(6)
- 资源 (2)
- 收藏
- 关注
RSS订阅原创 pytorch统计模型参数量
用resnet50 来举例子 print("resnet50 have {} paramerters in total".format(sum(x.numel() for x in resnet50.parameters()))) 其中numel表示含有多少element，通过此操作可以统计模型的参数量有多少 另外，两个是一样的，方便debug看其中到底有什么东西 ...
2018-08-24 14:36:00
6143
2
原创 pytorch使用tensorboardX进行网络可视化
我们知道，对于pytorch上的搭建动态图的代码的可读性非常高，实际上对于一些比较简单的网络，比如alexnet，vgg阅读起来就能够脑补它们的网络结构，但是对于比较复杂的网络，如unet，直接从代码脑补网络结构可能就比较吃力 tensorflow上的tensorboard进行计算图的可视化可谓是非常成熟了，那么有没有可以可视化pytorch动态图的工具呢？ 实际上是有的，前两天介绍了tens...
2018-08-17 22:26:00
414
原创 Vim删除文件到行首或者行尾
vim用的不是很熟练，只是有时候需要的时候会学习一下 我们知道，vim有三种模式，一种是一般模式，一种是编辑模式，另外一种是命令行模式 在一般模式下，可以进行删除，复制粘贴等操作，在编辑模式下可以编辑文件，在命令行模式下就是保存，读取文件了 现在我的问题背景是这样的：我想删除文件的某一行到文件的结尾（即删除文件的某一行到结尾所有的行），因为剩下的行都是我不需要的 看到了《鸟哥的linux私...
2018-08-14 10:44:00
3191
原创 pytorch使用tensorboardX进行loss可视化
最近pytorch出了visdom，也没有怎么去研究它，主要是觉得tensorboardX已经够用，而且用起来也十分的简单 pip install tensorboardX 然后在代码里导入 from tensorboardX import SummaryWriter 然后声明一下自己将loss写到哪个路径下面 writer = SummaryWriter('./lo...
2018-08-07 16:51:00
7558
2
原创 使用Python中的log模块将loss输出到终端与保存到文件
记得之前对深度学习中得loss输出，经常自己会将输出流重新定向到一个文件中， 比如 python main.py 2 > &1 | tee log.txt 对于caffe这种c++框架而言，用输出流重定向的方法是比较妥的，但是这样输出会有一个问题，比如，出了文件中print特定格式得loss之外， 你进行debug的时候，想print一些东西，debug print的...
2018-08-07 16:28:00
2733
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人