/Active-Shift-TF

Tensorflow implementation for Active Shift Layer(ASL)

Primary LanguageC++Apache License 2.0Apache-2.0

Active Shift Layer

This repository contains the implementation for Active Shift Layer (ASL).

Please see the paper Constructing Fast Network through Deconstruction of Convolution.

This paper is accepted in NIPS 2018 as spotlight session.

The code is based on Tensorflow
Caffe implementation is also available at ASL-Caffe

Prerequisite

Note that this code is tested only in the environment decribed below. Mismatched versions does not guarantee correct execution.

  • Tensorflow 1.4.1
  • Cuda 8.0
  • g++ 4.9.3

Build

  1. Edit CUDA_HOME in build.sh and build-spec for your GPU
  2. run >./build.sh
  3. If there is an error with cuda_config.h, run "cp ./lib/cuda_config.h $TF_INC/tensorflow/stream_executor/cuda/"

Testing

  1. run >python test_forward_ASL.py

    • It just shows the results of forwarding random tensor. You should not have any error message.

  2. run >python test_backward_ASL.py

    • You should get "OK" for 3 tests.

Usage

  1. import lib.active_shift2d_op as active_shift2d_op
  2. use active_shift2d_op.active_shift2d_op(input_tensor, shift_tensor, strides, paddings)
    • Please see test_forward_ASL.py