Source code for snapchat.research.gbml.trained_model_metadata_pb2
"""
@generated by mypy-protobuf. Do not edit manually!
isort:skip_file
"""
import builtins
import google.protobuf.descriptor
import google.protobuf.message
import sys
if sys.version_info >= (3, 8):
import typing as typing_extensions
else:
import typing_extensions
[docs]
class TrainedModelMetadata(google.protobuf.message.Message):
"""The path to a trained PyTorch model."""
"""The path to above model which is TorchScripted."""
"""The path where evaluation metrics are stored"""
"""Path where tensorboard logs will be stored"""
def __init__(
self,
*,
trained_model_uri: builtins.str = ...,
scripted_model_uri: builtins.str = ...,
eval_metrics_uri: builtins.str = ...,
tensorboard_logs_uri: builtins.str = ...,
) -> None: ...
[docs]
def ClearField(self, field_name: typing_extensions.Literal["eval_metrics_uri", b"eval_metrics_uri", "scripted_model_uri", b"scripted_model_uri", "tensorboard_logs_uri", b"tensorboard_logs_uri", "trained_model_uri", b"trained_model_uri"]) -> None: ...