update dllib-mlflow.py
Some checks failed
s444356-evaluation/pipeline/head This commit looks good
s444356-training/pipeline/head There was a failure building this commit

This commit is contained in:
Maciej Czajka 2022-05-09 22:08:18 +02:00
parent 2cb59732d1
commit b0c7aba70a

View File

@ -2,6 +2,7 @@ import numpy as np
import sys import sys
import os import os
import torch import torch
Import mlflow
import pandas as pd import pandas as pd
from torch import nn from torch import nn
from torch.autograd import Variable from torch.autograd import Variable