s434749/ium_434749
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

archvie results

Browse Source
This commit is contained in:
Alagris 2021-04-26 08:25:18 +02:00
parent 9cc09214cb
commit 8f958c000e
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train-model.Jenkinsfile
View File

@ -13,7 +13,7 @@ pipeline {
post { post {
success { success {
archiveArtifacts 'cnn.pth' archiveArtifacts 'cnn.pth,results.txt'
} }
} }
} }

2
train_model.py
View File

@ -165,7 +165,9 @@ def evaluate_monte_carlo(model, repeats):
i += BATCH_SIZE i += BATCH_SIZE
inner_bar.update(2 * BATCH_SIZE) inner_bar.update(2 * BATCH_SIZE)
outer_bar.update(1) outer_bar.update(1)
with open('results.txt') as r:
print("Average estimation error " + str(diff.item() / i)) print("Average estimation error " + str(diff.item() / i))
r.write("Average estimation error " + str(diff.item() / i)+"\n")
cnn = CNN(kernel_size=3, hidden_layers=14, channels=MAX_LEN, embedding_size=MAX_LEN).to(DEVICE) cnn = CNN(kernel_size=3, hidden_layers=14, channels=MAX_LEN, embedding_size=MAX_LEN).to(DEVICE)