s434766/ium_434766
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

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

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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') ]
}
}
}

1
pytorch-training-eval/Jenkinsfile-train
View File

@ -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'
}

2
sacred-pytorch1.py
View File

@ -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')

2
sacred-pytorch2.py
View File

@ -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')