Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network - View it on GitHub