lab is a Go package built on top of gotch - Pytorch deep learning framework for Go to facilitate deep learning research.
lab provides tool kits for common tasks of deep learning model training such as:
- dataloader
- data augmentation
- optimizer builder
- learning rate scheduler
- model builder
- evaluation metrics
- evaluator
- trainer
- loss tracker
- logger
- training configuration in yaml file
- metrics for segmentation and classification tasks
lab is also packed with data, draw and plot sub-packages.
This package is in progress and obviously there are many changes ahead. Feel free to contribute.