TorchFusion 是一个深度学习框架，主要用于 AI 系统加速研究和开发

TorchFusion 是一个深度学习框架，主要用于 AI 系统加速研究和开发。

TorchFusion 基于 PyTorch 并且完全兼容纯 PyTorch 和其他 PyTorch 软件包，它供了一个全面的可扩展训练框架，可以轻松用开发者的 PyTorch 模型进行训练，评估和运行推理。

该框架具有高度可扩展性，所以开发者可以根据自己特定的目的来进行训练。

Github 链接：

https://github.com/johnolafenwa/TorchFusion

  独特功能

高度可扩展

高度详细的汇总功能，不仅为您提供了有关参数，层数，输入和输出大小的详细信息，还为网络中的每个线性和卷积层提供了Flops（Multiply-Adds）的数量。 现在，只需一个功能就可以知道任何CNN架构的确切计算成本！

实时指标和损失可视化，并可选择永久保存它们

支持永久保存日志

易于使用的回调

注意：这只是 TorchFusion 的预发布版本，未来的 TorchFusion 将会跨越更多的深度学习领域。

  安装

安装 TorchFusion

pip3 install https://github.com/johnolaf