Searched defs:model_functorch (Results 1 – 1 of 1) sorted by relevance
16 model_functorch = convert_batchnorm_modules(models.resnet18(num_classes=10)) variable17 model_functorch = model_functorch.to(device) variable