SCALE-Sim 项目使用教程
SCALE-Sim 项目地址: https://gitcode.com/gh_mirrors/sc/SCALE-Sim
1. 项目目录结构及介绍
SCALE-Sim 项目的目录结构如下:
SCALE-Sim/
├── configs/
├── images/
├── topologies/
├── .gitignore
├── LICENSE
├── README.md
├── clean.sh
├── dram_trace.py
├── gen_min.py
├── getting_started.md
├── ideas.md
├── requirements.txt
├── run_nets.py
├── scale.py
├── sram_traffic_is.py
├── sram_traffic_os.py
├── sram_traffic_ws.py
└── trace_gen_wrapper.py
目录介绍:
- configs/:包含项目的配置文件,用于定义加速器的架构参数。
- images/:可能包含项目相关的图片或图表。
- topologies/:包含神经网络拓扑的描述文件,通常为 CSV 格式。
- .gitignore:Git 忽略文件,定义了哪些文件或目录不需要被 Git 管理。
- LICENSE:项目的开源许可证文件,通常为 MIT 许可证。
- README.md:项目的介绍文件,包含了项目的基本信息、使用方法和贡献指南。
- clean.sh:用于清理项目的脚本文件。
- dram_trace.py:与 DRAM 访问相关的脚本文件。
- gen_min.py:生成最小配置的脚本文件。
- getting_started.md:入门指南,帮助用户快速上手项目。
- ideas.md:可能包含项目的一些想法或未来计划。
- requirements.txt:项目依赖的 Python 包列表。
- run_nets.py:运行神经网络的脚本文件。
- scale.py:项目的启动文件,用于运行 SCALE-Sim 模拟器。
- sram_traffic_is.py、sram_traffic_os.py、sram_traffic_ws.py:与 SRAM 访问相关的脚本文件。
- trace_gen_wrapper.py:生成访问轨迹的包装脚本文件。
2. 项目的启动文件介绍
项目的启动文件是 scale.py
。该文件是 SCALE-Sim 模拟器的主要入口点,用户可以通过运行该文件来启动模拟器并执行神经网络的模拟。
使用方法:
python scale.py
用户可以通过命令行参数指定不同的配置文件和网络拓扑文件,例如:
python scale.py -arch_config=configs/eyeriss.cfg -network=topologies/yolo.csv
3. 项目的配置文件介绍
配置文件位于 configs/
目录下,通常为 .cfg
文件。这些文件定义了加速器的架构参数,如 SRAM 大小、阵列大小、接口带宽等。
配置文件示例:
[Architecture]
array_size = 16
sram_size = 1024
interface_bandwidth = 128
配置文件参数:
- array_size:定义了加速器中计算阵列的大小。
- sram_size:定义了加速器中 SRAM 的大小。
- interface_bandwidth:定义了加速器与外部存储器之间的接口带宽。
用户可以根据自己的需求创建新的配置文件,并通过命令行参数指定使用该配置文件。