[][src]Module tch::vision::resnet

ResNet implementation.

See "Deep Residual Learning for Image Recognition" He et al. 2015 https://arxiv.org/abs/1512.03385

Functions

resnet18

Creates a ResNet-18 model.

resnet18_no_final_layer
resnet34

Creates a ResNet-34 model.

resnet34_no_final_layer
resnet50
resnet50_no_final_layer
resnet101
resnet101_no_final_layer
resnet150_no_final_layer
resnet152