s426206/ium_426206
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

change addresses
All checks were successful
s426206-evaluation/pipeline/head This commit looks good
Details
s426206-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Jan Nowak 2021-05-23 16:42:51 +02:00
parent 60c251ae33
commit f30073b2ec
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
dlgssdpytorch.py
View File

@ -11,7 +11,7 @@ from sacred.observers import MongoObserver, FileStorageObserver
ex = Experiment("426206", interactive=False, save_git_info=False) ex = Experiment("426206", interactive=False, save_git_info=False)
ex.observers.append(FileStorageObserver('my_runs')) ex.observers.append(FileStorageObserver('my_runs'))
#ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred')) ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', db_name='sacred'))
class LayerLinearRegression(nn.Module): class LayerLinearRegression(nn.Module):
def __init__(self): def __init__(self):
super().__init__() super().__init__()

4
generate_MLmodel.py
View File

@ -29,8 +29,8 @@ if __name__ == "__main__":
lr = args.lr lr = args.lr
n_epochs = args.epochs n_epochs = args.epochs
mlflow.set_tracking_uri("http://127.0.0.1:5000") #mlflow.set_tracking_uri("http://127.0.0.1:5000")
#mlflow.set_tracking_uri("http://172.17.0.1:5000") mlflow.set_tracking_uri("http://172.17.0.1:5000")
mlflow.set_experiment("s426206") mlflow.set_experiment("s426206")
with mlflow.start_run(): with mlflow.start_run():
mlflow.log_param("lr", lr) mlflow.log_param("lr", lr)