s444354
/
ium_444354
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

signature should by numpy array
s444354-training/pipeline/head There was a failure building this commit Details

This commit is contained in:
Adrian Charkiewicz 2022-05-15 20:13:30 +02:00
parent 6b4ac85f8a
commit 28dbfa5d6f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch/pytorch.py
View File

@ -186,7 +186,7 @@ def fit(epochs, lr, model, train_loader, val_loader, _log, _run, opt_func=torch.
_run.info["epochs"] = epochs _run.info["epochs"] = epochs
signature = mlflow.models.signature.infer_signature(train_ds) signature = mlflow.models.signature.infer_signature(inputs_array)
tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme tracking_url_type_store = urlparse(mlflow.get_tracking_uri()).scheme