Module tch::vision::densenet

source ·
Expand description

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/tch-rs/releases/download/mw/densenet121.ot

Functions