pytorch

ed0f2733 - Minor edits in decorators to record time