Improve docstring of image classifier model spec.
PiperOrigin-RevId: 515466722
This commit is contained in:
parent
517e997179
commit
2d8f937913
|
@ -28,7 +28,7 @@ class ModelSpec(object):
|
|||
uri: str,
|
||||
input_image_shape: Optional[List[int]] = None,
|
||||
name: str = ''):
|
||||
"""Initializes a new instance of the `ImageModelSpec` class.
|
||||
"""Initializes a new instance of the image classifier `ModelSpec` class.
|
||||
|
||||
Args:
|
||||
uri: str, URI to the pretrained model.
|
||||
|
|
Loading…
Reference in New Issue
Block a user