fix
All checks were successful
s434749-training/pipeline/head This commit looks good

This commit is contained in:
Alagris 2021-05-23 20:31:41 +02:00
parent 1dda33aec7
commit 97e721bbdd

View File

@ -197,7 +197,7 @@ def signature(model, in_alphabet, max_len):
mock_x = [('abc', 'xyz')]
mock_text, _ = encode_str(mock_x, in_alphabet, max_len)
mock_y = model(mock_text)
return mlflow.models.signature.infer_signature(mock_text.numpy(), mock_y.numpy())
return mlflow.models.signature.infer_signature(mock_text.detach().numpy(), mock_y.detach().numpy())
@ex.automain