代码阅读:LanGCN

@toc<>

1训练

1.1 进度条

python 复制代码
import tqdm as tqdm
for i, data in tqdm(enumerate(train_loader),disable=hvd.rank()):

1.2 多进程通信

多线程通信依靠共享内存实现,但是多进程通信就麻烦很多,因此可以采用mpi库,如果是在python中使用,可以使用pmi4py

参考:Python多进程并行编程实践-mpi4py的使用

相关推荐
魔乐社区14 小时前
MiniCPM-V4.0开源并上线魔乐社区,多模态能力进化,手机可用,还有最全CookBook!
人工智能·深度学习·开源·大模型
飞机火车巴雷特14 小时前
【DeepResearch调研】基于知识图谱与数据合成的大语言模型幻觉缓解研究前沿
人工智能·大模型·知识图谱·数据合成·幻觉缓解
小饼干超人1 天前
【cs336学习笔记】[第6课]内核优化与Triton框架应用
深度学习·大模型·推理加速
小饼干超人2 天前
详解triton.jit及PTX
人工智能·大模型·推理加速
JolyouLu2 天前
LangChain4J-基础(整合Spring、RAG、MCP、向量数据库、提示词、流式输出)
大模型·提示词·向量数据库·rag·mcp·langchain4j
dundunmm2 天前
【每天一个知识点】训推一体机
人工智能·大模型·硬件·软件·训练·推理
地平线开发者2 天前
征程 6X | 常用工具介绍
算法·自动驾驶
地平线开发者2 天前
理想汽车智驾方案介绍 2|MindVLA 方案详解
算法·自动驾驶
地平线开发者2 天前
LLM 中评价指标与训练概要介绍
算法·自动驾驶
Godspeed Zhao2 天前
Tesla自动驾驶域控制器产品(AutoPilot HW)的系统化梳理
人工智能·机器学习·自动驾驶