Welcome to TorchTools’ documentation!

Helpful PyTorch utilities which should save you time—and heartache.

See the TorchTools repository on GitHub.

Contents:

• FCNet
  • FCNet
• ConvNet2d
  • ConvNet2d
• UNet
  • UNet
• Encoder2d
  • Encoder2d
• Decoder2d
  • Decoder2d
• AutoEncoder2d
  • AutoEncoder2d
• VAE2d
  • VAE2d
• SimpleConvNet2d
  • SimpleConvNet2d
• DataSet
  • DataSet
• ShapesDataset
  • ShapesDataset
• Misc
  • divides_by_two_check()
• Torch utils
  • disable_biases()
  • img_batch_dims_power_of_2()
  • patchify_img_batch()
  • target_from_mask_img()
  • total_image_variation()
• Weight initialisation
  • normal_init()
• File utilities
  • ls_zipfile()
  • traverse_directory_tree()

Indices and tables

• Index

• Module Index

• Search Page