Searched defs:aux_loss (Results 1 – 1 of 1) sorted by relevance
| /external/pytorch/benchmarks/functional_autograd_benchmark/ |
| D | torchvision_models.py | 453 arch_type, backbone, pretrained, progress, num_classes, aux_loss, **kwargs argument
470 pretrained=False, progress=True, num_classes=21, aux_loss=None, **kwargs argument
|