Skip to main content

No project description provided

Project description




PyTorchFI is a runtime perturbation tool for deep neural networks (DNNs), implemented for the popular PyTorch deep learning platform. PyTorchFI enables users to perform perturbation on weights or neurons of a DNN during runtime. It is extremely versatile for dependability and reliability research, with applications including resiliency analysis of classification networks, resiliency analysis of object detection networks, analysis of models robust to adversarial attacks, training resilient models, and for DNN interpertability.

An example of a use case for PyTorchFI is to simulate an error by performaing a fault-injection on an object recognition model.

Golden Output Output with Fault Injection


Download on PyPI here, or take a look at our documentation at

You can also learn more with our interactive demo.


From Pip

Install using pip install pytorchfi

From Source

Download this repository into your project folder.


Import the entire package:

import pytorchfi

Import a specific module:

from pytorchfi import core





The main source code of PyTorchFI is held in pytorchfi, which carries both core and implementations.


All python code is formatted with black.


Before contributing, please refer to our contributing guidelines.


View the published paper. If you use or reference PyTorchFI, please cite:

author={A. {Mahmoud} and N. {Aggarwal} and A. {Nobbe} and J. R. S. {Vicarte} and S. V. {Adve} and C. W. {Fletcher} and I. {Frosio} and S. K. S. {Hari}},
booktitle={2020 50th Annual IEEE/IFIP International Conference on Dependable Systems and Networks Workshops (DSN-W)},
title={PyTorchFI: A Runtime Perturbation Tool for DNNs},

Funding Sources

This project was funded in part by the Applications Driving Architectures (ADA) Research Center, a JUMP Center co-sponsored by SRC and DARPA, and in collaboration with NVIDIA Research.


NCSA License. Copyright © 2021 RSim Research Group.

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

pytorchfi-0.6.0.tar.gz (16.6 kB view hashes)

Uploaded source

Built Distribution

pytorchfi-0.6.0-py3-none-any.whl (21.2 kB view hashes)

Uploaded py3

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page