changed cpu count function

Docker/Image/code/main.py

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