
代码优化
文章平均质量分 71
兢兢业业小码农
深度学习+python+有意思的小东西
分享爬坑经验
展开
-
python代码性能优化
1.可视化代码运行时间工具vprof:https://github.com/nvdv/vprof安装:sudo pip3 install vprof然后直接用它运行代码:vprof -c h test.pyh会让它根据每行代码的运行时间附上热图。需要带输入时:vprof -c cmh "testscript.py --foo --bar"2.用上下文管理器测量部分代码运行时间:from time ...转载 2018-06-15 17:31:03 · 1844 阅读 · 0 评论 -
pytorch多进程加速及代码优化
目标:优化代码,利用多进程,进行近实时预处理、网络预测及后处理:本人尝试了pytorch的multiprocessing,进行多进程同步处理以上任务。from torch.multiprocessing import Pool,Manager为了进行各进程间的通信,使用Queue,作为数据传输载体。 manager = Manager() input_queue = manager.Q...原创 2018-06-20 16:26:22 · 16913 阅读 · 10 评论 -
python3 dict ndarray 存成json,并保留原数据精度
import numpy as npimport codecs, json a = np.arange(10).reshape(2,5) # a 2 by 5 arrayb = a.tolist() # nested lists with same data, indicesfile_path = "/path.json" ## your path variablejson.dump...转载 2018-08-31 09:38:19 · 5866 阅读 · 0 评论