- Helps you avoid lots of boilerplate code you write every time from scratch while training a model in pytorch.
- Easy to use deepml API.
- Supports Semantic segmentation.
- You can also define your custom machine learning task. deepml library is extensible.
- With 3-5 lines of deepml API you can start training your model in pytorch:
from deepml.tasks import ImageClassification
from deepml.train import Learner
task = ImageClassification(model, 'experiment1')
learner = Learner(ml_task, optimizer, criterion)
learner.fit(train_loader, val_loader)
What's Changed
* Feature/multi predictor by sagar-rathod in https://github.com/sagar-rathod/deep-ml/pull/11
* Feature/multi predictor by sagar-rathod in https://github.com/sagar-rathod/deep-ml/pull/12
* Update notebook links in README by sagar-rathod in https://github.com/sagar-rathod/deep-ml/pull/13
**Full Changelog**: https://github.com/sagar-rathod/deep-ml/compare/v1.1.0...v2.0.0