Pytorch Wrapper v1.1

Hey all,

PyTorch Wrapper version 1.1 is out!

New Features:

  • Samplers for smart batching based on text length for faster training.
  • Loss and Evaluation wrappers for token prediction tasks.
  • New nn.modules for attention based models.
  • Support for multi GPU training / evaluation / prediction.
  • Verbose argument in system’s methods.
  • Examples using Transformer based models like BERT for text classification.

Check it out in the following links:

install with: pip install pytorch-wrapper

#DeepLearning #PyTorch #NeuralNetworks #MachineLearning #DataScience #python #tensor

1 Like