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
GitHub: https://github.com/jkoutsikakis/pytorch-wrapper 5
docs: https://pytorch-wrapper.readthedocs.io/en/latest/ 2
examples: https://github.com/jkoutsikakis/pytorch-wrapper/tree/master/examples 2
#DeepLearning #PyTorch #NeuralNetworks #MachineLearning #DataScience #python #tensor