mirror of
https://github.com/wassname/ray.git
synced 2026-10-04 13:00:39 +08:00
[Serve] Fix PyTorch Tutorial (#9019)
This commit is contained in:
1 parent
92f67cd2ae
commit
e89b8d37ae
1 file changed
+3
-1
@@ -16,12 +16,14 @@ from torchvision.models import resnet18
|
||||
# __doc_define_servable_begin__
|
||||
class ImageModel:
|
||||
def __init__(self):
|
||||
self.model = resnet18(pretrained=True)
|
||||
self.model = resnet18(pretrained=True).eval()
|
||||
self.preprocessor = transforms.Compose([
|
||||
transforms.Resize(224),
|
||||
transforms.CenterCrop(224),
|
||||
transforms.ToTensor(),
|
||||
transforms.Lambda(lambda t: t[:3, ...]), # remove alpha channel
|
||||
transforms.Normalize(
|
||||
mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]),
|
||||
])
|
||||
|
||||
def __call__(self, flask_request):
|
||||
|
||||
Reference in new issue
Block a user