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