Skip to main content

Library for training deep neural nets in Pytorch.

Project description

PytorchDeepML

This library is a wrapper around pytorch and useful for solving image classification and semantic segmentation problems.

Features

  1. Easy to use wrapper around pytorch so that you can focus on training and validating your model.

  2. Integrates with Tensorboard to use it to monitor metrics while model trains.

  3. Quickly visualize your model's predictions.

Installation

Before installing deepml, it is recommended to refer pytorch official page for torch installation.

Pypi

pip install deepml

Examples

Check out the below google colaboratory notebook examples:

  1. Image Regression.
  2. Image Classification.
  3. Semantic Segmentation. [Coming Soon]

Contributing

If you find a bug or have a feature request, file an issue at https://github.com/sagar-rathod/PytorchDeepML/issues

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

deepml-1.1.0.tar.gz (133.6 kB view hashes)

Uploaded Source

Built Distribution

deepml-1.1.0-py3-none-any.whl (140.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page