model_resnet18: ResNet-18 Model Architecture

Description Usage Arguments See Also

View source: R/models-resnet.R

Description

ResNet-18 model architecture from Deep Residual Learning for Image Recognition

Usage

1
model_resnet18(pretrained = FALSE, progress = TRUE, ...)

Arguments

pretrained

(bool): If TRUE, returns a model pre-trained on ImageNet.

progress

(bool): If TRUE, displays a progress bar of the download to stderr.

...

Other parameters passed to the resnet model.

See Also

Other models: model_alexnet()


torchvision documentation built on Aug. 17, 2021, 5:06 p.m.