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