摘要: 深度学习骨干网络一般会包含很多层,这里写了一个脚本,可以保存骨干网络的所有特征图。 代码主要用了get_graph_node_names和create_featrue_extractor这两个函数。 get_graph_node_names是得到所有特征节点名字。 create_featrue_ex 阅读全文
posted @ 2024-09-30 22:31 Dsp Tian 阅读(62) 评论(0) 推荐(0) 编辑
摘要: 比如nvidia-smi命令会输出当前时刻显卡占用率,如果想动态监控,可以输入如下命令: watch -n 0.1 nvidia-smi 阅读全文
posted @ 2024-09-30 17:42 Dsp Tian 阅读(16) 评论(0) 推荐(0) 编辑