Skip to main content

Nod AutoML package for neural architecture search and model training of image classification

Project description

NodLabsNNS pip package

Nod Labs AutoML package for neural architecture search and model training of image classification

Requirements

Specific steps

  1. install pip package
pip3 install nod-labs-NNS==0.0.1
  1. Setting up ImageNet dataset

To setup the ImageNet follow the instructions from here

  1. Use nodlabsNNS package for neural architecture search and train

create a python file

Search

import nns

model = model.nod_detection('tpu','your-tpu-name','your-bucket-name')
model.dataset('imagenet')
model.search_model()

Train

import nns

model = model.nod_detection('tpu','your-tpu-name','your-bucket-name')
model.dataset('imagenet')
model.train_model()

Currently nns only support TPU for searching and training and only support IMAGENET dataset.

Project details


Download files

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

Files for nod-labs-NNS, version 0.0.1
Filename, size File type Python version Upload date Hashes
Filename, size nod_labs_NNS-0.0.1-py3-none-any.whl (42.0 kB) File type Wheel Python version py3 Upload date Hashes View
Filename, size nod-labs-NNS-0.0.1.tar.gz (36.3 kB) File type Source Python version None Upload date Hashes View

Supported by

AWS AWS Cloud computing Datadog Datadog Monitoring Facebook / Instagram Facebook / Instagram PSF Sponsor Fastly Fastly CDN Google Google Object Storage and Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Salesforce Salesforce PSF Sponsor Sentry Sentry Error logging StatusPage StatusPage Status page