资源分类
》 资源 》算法》online-softmax

online-softmax

2019-12-24 | |  42 |   0 |   0

Benchmark

This is a benchmark to explore performance of Softmax and Softmax+TopK functions with online calculation of softmax normalization term. Please see "Online normalizer calculation for softmax" paper.

Build

You will need CUB v1.8.0 (or newer) and CUDA 9.1 (or newer) to build the benchmark. Set CUB variable in Makefile to the correct location of CUB library. Build the sample: make

Run

Run the sample: ./online_softmax_benchmark


上一篇：mxnet_to_onnx

下一篇：gpu-rest-engine

用户评价
全部评价

热门资源

  • seetafaceJNI

    项目介绍 基于中科院seetaface2进行封装的JAVA...

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

    This is a very basic boilerplate example for pe...