s464863/ium_464863
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

IUM_07 - fix models/Jenkinsfile, update create_model.py

Browse Source
This commit is contained in:
Paweł Łączkowski 2024-04-24 22:01:29 +02:00
parent b2c46dcd54
commit 93a4420755
2 changed files with 1 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
create_model.py
View File

@ -33,12 +33,6 @@ def config():
weight_decay = 0.001
num_epochs = 1000
# Learning parameters from sys.argv
if len(sys.argv) > 1:
num_epochs = int(sys.argv[1])
learning_rate = float(sys.argv[2])
weight_decay = float(sys.argv[3])
@ex.automain
def experiment(hidden_size, learning_rate, weight_decay, num_epochs, _run):
# Seed for reproducibility

2
models/Jenkinsfile vendored
View File

@ -50,7 +50,7 @@ pipeline {
steps {
script {
sh "chmod +x ./create_model.py"
sh "python3 ./create_model.py with 'epochs=${params.epochs}' 'learning_rate=${params.learning_rate}' 'weight_decay=${params.weight_decay}'"
sh "python3 ./create_model.py with 'num_epochs=${params.epochs}' 'learning_rate=${params.learning_rate}' 'weight_decay=${params.weight_decay}'"
def experiment_id = readFile('experiment_id.txt').trim()
archiveArtifacts artifacts: "my_runs/${experiment_id}/*", onlyIfSuccessful: true