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


Release history Release notifications

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

Pingdom Pingdom Monitoring Google Google Object Storage and Download Analytics Sentry Sentry Error logging AWS AWS Cloud computing DataDog DataDog Monitoring Fastly Fastly CDN DigiCert DigiCert EV certificate StatusPage StatusPage Status page