s487194/ium_487194
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Zaktualizuj 'train.py'

Browse Source
This commit is contained in:
Witold Woch 2023-05-14 23:23:37 +02:00
parent 15bee98414
commit e394fdca51
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train.py
View File

@ -102,7 +102,7 @@ with mlflow.start_run() as run:
loss.backward() loss.backward()
optimizer.step() optimizer.step()
# Infer model signature to log it torch.save(model,"classificationn_model.pt")
signature = infer_signature(X_train.numpy(), model(X_train).detach().numpy()) signature = infer_signature(X_train.numpy(), model(X_train).detach().numpy())
input_example = {"input": X_train[0].numpy().tolist()} input_example = {"input": X_train[0].numpy().tolist()}