Imported from https://github.com/metalbubble/TRN-pytorch