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,
|
uri: str,
|
||||||
input_image_shape: Optional[List[int]] = None,
|
input_image_shape: Optional[List[int]] = None,
|
||||||
name: str = ''):
|
name: str = ''):
|
||||||
"""Initializes a new instance of the `ImageModelSpec` class.
|
"""Initializes a new instance of the image classifier `ModelSpec` class.
|
||||||
|
|
||||||
Args:
|
Args:
|
||||||
uri: str, URI to the pretrained model.
|
uri: str, URI to the pretrained model.
|
||||||
|
|
Loading…
Reference in New Issue
Block a user