Module tch::vision::densenet

DenseNet implementation.

See "Densely Connected Convolutional Networks", Huang et al 2016. https://arxiv.org/abs/1608.06993

Pre-trained weights for the densenet-121 variants can be found here: https://github.com/LaurentMazare/ocaml-torch/releases/download/v0.1-unstable/densenet121.ot

Functions

densenet121
densenet161
densenet169
densenet201