sytelus / archai_old

Official Implementation of 'Fast AutoAugment' in PyTorch.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ArchAI Library

Installation

  • Ubuntu
  • Install cuda 10.1
  • Install anaconda package manager
  • conda env create -f archaicuda101.yml
  • conda activate archaicuda101
  • pip install -r requirements.txt
  • pip install -e .

Test installation

  • From archai root 'bash scripts/test_toy_examples/test_toy_examples.sh'

About

Official Implementation of 'Fast AutoAugment' in PyTorch.

License:MIT License


Languages

Language:Python 99.4%Language:Dockerfile 0.4%Language:Shell 0.3%