Module: TorchVision::Models::ResNet18

Defined in:
lib/torchvision/models/resnet18.rb

Class Method Summary collapse

Class Method Details

.new(**kwargs) ⇒ Object 



4
5
6
 
# File 'lib/torchvision/models/resnet18.rb', line 4

def self.new(**kwargs)
  ResNet.make_model("resnet18", BasicBlock, [2, 2, 2, 2], **kwargs)
end