
TensorFlow-Framework-for-Implementing-DNN-with-Memristor-Arrays...
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文探讨了使用忆阻器阵列在TensorFlow框架中实现深度神经网络(DNN)的方法,展示了该技术在网络加速与硬件集成方面的潜力。
我们在此代码中为基于逆变器的忆阻神经形态硬件设计了一种改进的培训框架,以便研究社区更易于使用。利用行业标准的TensorFlow工具,在先前的Matlab训练范例基础上,我们创建了一个声明式的编程方案来实现非现场训练。通过有向图表示和计算神经网络训练过程,该框架能够支持更加并行且高效的计算。
文件说明:关于此项目的详细报告以IEEE格式编写,并可在最终报告pdf中查阅。“代码实现”文件夹包含以下内容:
- Tensorflow_DNN.py: 适用于MNIST数据集的TensorFlow训练框架
- memristor_DNN.py: 使用Numpy数组进行MNIST数据集训练的框架
- MNIST_complete.mat:上述网络培训所用到的MNIST数据集
先决条件:本培训框架需要以下模块:
- Python(版本2.7X或3.6X均可正常工作)
- TensorFlow
全部评论 (0)
还没有任何评论哟~


