Imported from https://github.com/marty1885/sparsenet-pytorch