module TorchVision::Models::ResNet18

Public Class Methods

new(**kwargs) click to toggle source
# File lib/torchvision/models/resnet18.rb, line 4
def self.new(**kwargs)
  ResNet.make_model("resnet18", BasicBlock, [2, 2, 2, 2], **kwargs)
end