Zaktualizuj 'ml_pytorch_mlflow.py'
This commit is contained in:
parent
6bceb78141
commit
311064bed4
@ -147,10 +147,9 @@ def my_main(epochs):
|
|||||||
input_, target = val_ds[i]
|
input_, target = val_ds[i]
|
||||||
file.write(str(predict_single(input_, target, model)))
|
file.write(str(predict_single(input_, target, model)))
|
||||||
|
|
||||||
val_ds_x = val_ds.unsqueeze(0)
|
|
||||||
input_example = val_ds_x[0]
|
input_example = val_ds_x[0]
|
||||||
# input_example = input_example.unsqueeze(0)
|
# input_example = input_example.unsqueeze(0)
|
||||||
signature = mlflow.models.signature.infer_signature(input_example, model(input_example))
|
signature = mlflow.models.signature.infer_signature(input_[0], prediction(input_[0]), model)
|
||||||
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
|
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme
|
||||||
|
|
||||||
if tracking_url_type_store != "file":
|
if tracking_url_type_store != "file":
|
||||||
|
Loading…
Reference in New Issue
Block a user