Commit 5371251b authored by Patrick Michalsky's avatar Patrick Michalsky
Browse files

changed cpu count function

parent ad7c67f8
from __future__ import print_function
from multiprocessing import cpu_count
import tensorflow as tf
import os
......@@ -49,7 +48,7 @@ with open('./out/accuracy.txt', 'a+') as f:
f.write(
'\n' +
'Number of GPUs: ' + str(gpu_count) + '\n' +
'Number of CPUs: ' + str(cpu_count()) + '\n' +
'Number of CPUs: ' + str(len(list(os.sched_getaffinity(0)))) + '\n' +
str('-' * 32) + '\n' +
'Accuracy: ' + str("%.2f" % round(training_history.history['accuracy'][-1] * 100, 2)) + '%\n\n'
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment