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