MyCaffe:带强化学习的Caffe完整C#重写
神经与进化计算
2018-10-05 v1
摘要
过去几年中,伯克利人工智能研究的Caffe在深度学习社区获得了大量追随者,在github.com/BLVC/Caffe站点上有超过15K个fork。凭借其组织良好、高度模块化的C++设计,它易于使用且速度很快。然而,在Windows开发领域,C#借助相较C++的诸多增强特性(如垃圾回收、丰富的.NET编程框架以及通过Entity Framework实现的简便数据库访问)加速了开发。那么C#开发者如何利用C#的进步来充分享用伯克利Caffe深度学习系统带来的益处?答案是面向Windows .NET程序员的完全开源的“MyCaffe”。MyCaffe是一个开源的、对伯克利Caffe的完整C#语言重写。本文描述了MyCaffe的总体架构,包括新加入的用于强化学习的MyCaffeTrainerRL。此外,本文讨论了MyCaffe如何在与C++ Caffe紧密对齐的同时,高效调用底层NVIDIA CUDA硬件,从而为Windows .NET程序员提供高性能、高可编程性的深度学习系统。
引用
@article{arxiv.1810.02272,
title = {MyCaffe: A Complete C# Re-Write of Caffe with Reinforcement Learning},
author = {David W. Brown},
journal= {arXiv preprint arXiv:1810.02272},
year = {2018}
}
备注
13 pages, 9 figures