TorchKit: A PyTorch Toolkit for Research

torchkit is a lightweight library containing PyTorch utilities useful for day-to-day research. Its main goal is to abstract away a lot of the redundant boilerplate associated with research projects like experimental configurations, logging and model checkpointing.

User Guide

• Installation

API Reference

• torchkit.logger
• torchkit.checkpoint
• torchkit.experiment
• torchkit.layers
• torchkit.losses
• torchkit.utils
  • torchkit.utils.config
  • torchkit.utils.dataset
  • torchkit.utils.git
  • torchkit.utils.module_stats
  • torchkit.utils.pdb_fallback
  • torchkit.utils.io
  • torchkit.utils.seed
  • torchkit.utils.timer

Indices and tables

• Index

• Module Index

• Search Page