diff --git a/Jenkinsfile b/Jenkinsfile index 6920eb4..f363f08 100644 --- a/Jenkinsfile +++ b/Jenkinsfile @@ -32,4 +32,9 @@ pipeline { } } } + post { + success { + build job: 's434766-training/master', parameters: [string(name: 'BATCH_SIZE', value:'16'), string(name: 'EPOCHS', value:'5') ] + } + } } \ No newline at end of file diff --git a/pytorch-training-eval/Jenkinsfile-train b/pytorch-training-eval/Jenkinsfile-train index 9fbae7d..a240979 100644 --- a/pytorch-training-eval/Jenkinsfile-train +++ b/pytorch-training-eval/Jenkinsfile-train @@ -43,6 +43,7 @@ pipeline { } post { success { + build job: 's434766-evaluation/master' mail body: 'SUCCESS TRAINING', subject: 's434766', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms' } diff --git a/sacred-pytorch1.py b/sacred-pytorch1.py index f359c24..f3d18c2 100644 --- a/sacred-pytorch1.py +++ b/sacred-pytorch1.py @@ -72,7 +72,7 @@ def train(num_epochs, batch_size, learning_rate, _log): info_loss = "Last loss = " + str(loss.item()) _log.info(info_loss) y_pred = model(fTest) - print("predicted Y value: ", y_pred.data) + # print("predicted Y value: ", y_pred.data) torch.save(model.state_dict(), 'stroke.pth') diff --git a/sacred-pytorch2.py b/sacred-pytorch2.py index 8ba9322..98848e7 100644 --- a/sacred-pytorch2.py +++ b/sacred-pytorch2.py @@ -73,7 +73,7 @@ def train(num_epochs, batch_size, learning_rate, _log): info_loss = "Last loss = " + str(loss.item()) _log.info(info_loss) y_pred = model(fTest) - print("predicted Y value: ", y_pred.data) + # print("predicted Y value: ", y_pred.data) torch.save(model.state_dict(), 'stroke.pth')