About Anaconda Help Download Anaconda

sahni / packages / torchutils 0.0.4

PyTorch utility APIs.

Installers

  • linux-64 v0.0.4
  • osx-64 v0.0.4
  • win-64 v0.0.4

conda install

To install this package run one of the following:
conda install sahni::torchutils

Description

TorchUtils is a Python package providing helpful utility APIs for your PyTorch projects.

Features:

  • Save/load checkpoints.
  • Calculate dataset statistics (mean, std, var). Also calculate and track running statistics of data.
  • Get/set learning rate.
  • Track evaluation metrics such as accuracy, running loss, hamming loss.
  • Print model summary. Supports: Linear/MLP, Convolution Network, Recurrent Network (RNN/LSTM/GRU), Recursive Network.
  • Calculate model FLOPs.
  • Calculate total model parameters.
  • Set random seed.
  • Visualize gradient flow in your network.

Requirements:

  • PyTorch >= 1.0.0
  • Numpy >= 1.16.2
  • Matplotlib >= 3.0.3

© 2025 Anaconda, Inc. All Rights Reserved. (v4.2.0) Legal | Privacy Policy