final sacred
All checks were successful
s434766-evaluation/pipeline/head This commit looks good
s434766-training/pipeline/head This commit looks good

This commit is contained in:
s434766 2021-05-13 23:32:59 +02:00
parent 54cd87708b
commit 866186c16e
4 changed files with 8 additions and 2 deletions

5
Jenkinsfile vendored
View File

@ -32,4 +32,9 @@ pipeline {
} }
} }
} }
post {
success {
build job: 's434766-training/master', parameters: [string(name: 'BATCH_SIZE', value:'16'), string(name: 'EPOCHS', value:'5') ]
}
}
} }

View File

@ -43,6 +43,7 @@ pipeline {
} }
post { post {
success { success {
build job: 's434766-evaluation/master'
mail body: 'SUCCESS TRAINING', subject: 's434766', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms' mail body: 'SUCCESS TRAINING', subject: 's434766', to: '26ab8f35.uam.onmicrosoft.com@emea.teams.ms'
} }

View File

@ -72,7 +72,7 @@ def train(num_epochs, batch_size, learning_rate, _log):
info_loss = "Last loss = " + str(loss.item()) info_loss = "Last loss = " + str(loss.item())
_log.info(info_loss) _log.info(info_loss)
y_pred = model(fTest) 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') torch.save(model.state_dict(), 'stroke.pth')

View File

@ -73,7 +73,7 @@ def train(num_epochs, batch_size, learning_rate, _log):
info_loss = "Last loss = " + str(loss.item()) info_loss = "Last loss = " + str(loss.item())
_log.info(info_loss) _log.info(info_loss)
y_pred = model(fTest) 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') torch.save(model.state_dict(), 'stroke.pth')