资源分类
》 资源 》算法》lgamma

lgamma

2019-09-19 | |  109 |   0 |   0

lgamma

Implementations of polygamma, lgamma, and beta functions for PyTorch. It's very hacky, but that's usually ok for research use.

To build, run:

./make.sh

You'll probably need to pass in the correct CUDA path to build.py, which is run inside make.sh, so modify it to instead call

python build.py --cuda-path YOUR_CUDA_PATH

Also, you'll probably need to change the architecture version/CUDA compute capability inside make.sh, so replace sm_35 with whatever your GPU supports. Feel free to open an issue if you run into another problem!

上一篇：flow

下一篇：mia

用户评价
全部评价

热门资源

  • Keras-ResNeXt

    Keras ResNeXt Implementation of ResNeXt models...

  • seetafaceJNI

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

  • spark-corenlp

    This package wraps Stanford CoreNLP annotators ...

  • capsnet-with-caps...

    CapsNet with capsule-wise convolution Project ...

  • inferno-boilerplate

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