Code Monkey home page Code Monkey logo

asymcdc's Introduction

AsymCDC

Repository structure

  • model: Code for training and distilling the model
  • clipper-asymcc: A prediction serving system that sits between user-facing applications and machine learning models and frameworks.It is built atop Clipper

Train a Deployed Model

The deployed model is based on i-RevNet. To train a deployed model, just use the following commands. Make sure that you have installed PyTorch and CUDA in advance.

cd model/src/train

python CIFAR_main.py --nBlocks 18 18 18 --nStrides 1 2 2 --nChannels 16 64 256

Distill a Decoder

The knowledge distillation is based on RepDistiller. To distill a decoder, just use the following commands.

cd model/src/distill

python train_decoder.py --path_t ./save/models/iRevNet18/speech_irevnet_t.t7 \
                        --dataset cifar10 \
                        --distill kd \
                        --model_s iRevNet16x64 \
                        --learning_rate 0.01 \
                        -r 0.1 \
                        -a 0.9 \
                        -b 0 \
                        --trial 1 \
                        --irev \
                        --epochs 210 \
                        --lr_decay_epochs 120,150,180 \
                        --ec_k 2 \
                        --batch_size 128 \
                        --et

Run Cloud Latency Experiments

First, you should have several nodes, each of which is recommended to be equipped with at least a GPU and is enabled with CUDA>=11.4.

Environmet

Users can run setup.sh script on each node to satisfy Clipper's software requirements.

cd clipper-asymcc/run
bash scripts/setup.sh

Start Cluster

Users can run clipper_deploy.sh on each node to start clipper cluster

python clipper_deploy.py

Run Experiments

You can set the configuration parameters in config/*.json at first. Then run the following commands to run experiments.

python run_exp.py --conf /path/to/your/config --path /path/to/your/file

asymcdc's People

Contributors

wl83 avatar

Watchers

 avatar

Forkers

wl83

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    ๐Ÿ–– Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. ๐Ÿ“Š๐Ÿ“ˆ๐ŸŽ‰

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google โค๏ธ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.