- 博客(3)
- 收藏
- 关注
原创 tmux使用
ssh链接不稳定,若直接在ssh终端中运行某个长时间的程序,会被中断。 使用tmux,即使ssh服务中断,tmux中的程序依旧运行着。
2022-07-18 11:36:30
368
原创 数据并行加载在多个gpu上
一般数据读取会涉及两个函数:1.get_dataset(data_path,cfg)。其中的__getitem__(self,idx)函数,这个函数用于读取给定路径下的数据并处理成所需的格式,返回一个数据的data,data一般为字典格式。作用:规范数据,path-->tensor(data)。2.dataloader(get_dataset),可直接调用pytorch中的接口。通过for i ,data_batch in enumerate(dataloader)调用前文提到的_.
2020-08-07 14:14:16
2154
原创 深度学习网络中设置部分层参数可调
1.基于框架pytorch。先查看网络每层的名字p[0],将需要梯度更新的层的名字记下,替换以下代码中的’conv5_ft.conv.weight/bias’,将不需要梯度更新层的require_grad设为False。for p in net.named_parameters():if p[0] != ‘conv5_ft.conv.weight’ and p[0]!=‘conv5_ft.co......
2019-05-29 09:34:51
775
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人