首页
学习
活动
专区
工具
TVP
发布

JNing的专栏

专栏作者
694
文章
965887
阅读量
50
订阅数
【pytorch】固定(freeze)住部分网络
因为:即使对bn设置了 requires_grad = False ,一旦 model.train() ,bn还是会偷偷开启update( model.eval()模式下就又停止update )。(详见【pytorch】bn) 所以:train每个epoch之前都要统一重新定义一下这块,否则容易出问题。
JNingWei
2021-12-06
1.9K0
【pytorch】onnx
pytorch任意形式的model(.t7、.pth等等)转.onnx全都可以采用固定格式。
JNingWei
2021-12-06
6360
【pytorch】tnn
首先确保已经安装了tnn环境:onnx2tnn 转tnn: def onnx2tnn(self): assert os.path.isfile(simplified_onnx_path) os.system('cd {}\n' 'python onnx2tnn.py {} -version=v1.0 -optimize=1 -half=1 -o {} -input_shape input:1,{},{},{}' .format(C.o
JNingWei
2021-12-06
2530
【visdom】配置visdom
如果pytorch训练进程在服务器上,而本地又想同步查看visdom,则需要同时在本地和服务器执行上面的指令。 建议把指令放到screen里面去执行,这样不会被误kill。
JNingWei
2020-03-18
3550
深度学习: 深度学习框架 横向对比
[1] TensorFlow和Caffe、MXNet、Keras等其他深度学习框架的对比
JNingWei
2018-09-28
9950
pytorch: 框架介绍
Tutorial pytorch.org/tutorials 莫烦Python-Pytorch教程系列 Org pytorch.org Community github.com/pytorch Mod
JNingWei
2018-09-27
9270
pytorch: tensor类型的构建与相互转换
其中,torch.Tensor、torch.rand、torch.randn 均默认生成 torch.FloatTensor型 :
JNingWei
2018-09-27
1.8K0
pytorch: PyTorch中 使用 Tensorboard
因为工作需要,改用pytorch。但如何将训练过程可视化成了大问题。听说pytorch代码中可以插入tensorboard代码,第一反应是居然可以这么玩。。
JNingWei
2018-09-27
3.3K0
没有更多了
社区活动
Python精品学习库
代码在线跑,知识轻松学
热点技术征文第五期
新风口Sora来袭,普通人该如何把握机会?
博客搬家 | 分享价值百万资源包
自行/邀约他人一键搬运博客,速成社区影响力并领取好礼
技术创作特训营·精选知识专栏
往期视频·干货材料·成员作品·最新动态
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档