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