tensorpack

收录理由

Tensorpack 是一款面向图模式 TensorFlow 的训练接口，它不把精力花在包装模型上，而是专注把训练速度提上去。它自带了多卡和分布式数据并行的现成方案，同时用 dataflow 机制把纯 Python 的数据预处理自动并行化，避免数据读取拖后腿——项目文档里也明确对比了 tf.data，认为这种取舍更适合科研中的灵活需求。仓库里还收录了多篇论文的可复现实现，不是 MNIST 之类的玩具示例，做研究的人可以直接拿来做 baseline。如果你在用 TensorFlow，又同时在意速度和灵活性，这个项目已经相当成熟，值得一试。