FAME is a fast and low memory cost simulation framework for non-ideal ReRAM-based DNN accelerator.
- Python >= 3.8
- PyTorch >= 2.0.1
- Matplotlib
- Set your dataset root path and result saving path in
config.ini
. - Simply run the following command:
to execute a noise injection training sample for LeNet-5 on FAME.
$ python mnist.py