torchsharp is a framework for PyTorch which provides a set of sharp utilities aiming at speeding up programming and encouraging code re-use.
This repository consists of:
-
torchsharp.data : Useful stuff about data operation such as dummy datasets and image transforms for data argumentation, etc.
-
torchsharp.model : Helpful fucntions about model training process such as initializer and metrics, etc.
- configuration profile
- high level model class
- weight initializer wrapper
- optimizer wrapper
- lr scheduler wrapper
- more metrics
- history and tensorboard logger
-
torchsharp.utils : Other tools like logger and timer, etc.