Constructs a ResNet-18 model. Args: pretrained (bool): If True, returns a model pre-trained on ImageNet
(pretrained=True, **kwargs)
| 153 | |
| 154 | |
| 155 | def resnet18(pretrained=True, **kwargs): |
| 156 | """Constructs a ResNet-18 model. |
| 157 | Args: |
| 158 | pretrained (bool): If True, returns a model pre-trained on ImageNet |
| 159 | """ |
| 160 | model = ResNet(BasicBlock, [2, 2, 2, 2], **kwargs) |
| 161 | return model |
| 162 | |
| 163 | |
| 164 | def resnet34(pretrained=True, **kwargs): |