module TorchVision::Models::WideResNet50_2

Public Class Methods

new(**kwargs) click to toggle source
# File lib/torchvision/models/wide_resnet50_2.rb, line 4
def self.new(**kwargs)
  kwargs[:width_per_group] = 64 * 2
  ResNet.make_model("wide_resnet50_2", Bottleneck, [3, 4, 6, 3], **kwargs)
end