OVERVIEW

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.