Module: TorchVision::Models::WideResNet50_2

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

Class Method Summary collapse

Class Method Details

.new(**kwargs) ⇒ Object 



4
5
6
7
 
# 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