fixed resnet50 config and sleep issue (#96)

argonne-lcf · Oct 6, 2023 · fb7cb86 · fb7cb86
1 parent 4e0995a
commit fb7cb86
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/dlio_benchmark/configs/workload/resnet50.yaml b/dlio_benchmark/configs/workload/resnet50.yaml
@@ -14,7 +14,7 @@ dataset:
  format: tfrecord
 
 train:
- computation_time: 0.01
+ computation_time: 0.098
 
 reader:
  data_loader: tensorflow

diff --git a/dlio_benchmark/framework/tf_framework.py b/dlio_benchmark/framework/tf_framework.py
@@ -17,7 +17,7 @@
 
 import os
 import logging
-from time import time
+from time import time, sleep
 
 from dlio_benchmark.common.constants import MODULE_AI_FRAMEWORK
 from dlio_benchmark.data_loader.data_loader_factory import DataLoaderFactory
@@ -112,7 +112,8 @@ def checkpoint(self, epoch, step_number):
 
     @dlp.log
     def compute(self, epoch_number, step, computation_time):
-        tf.function(self.model)(epoch_number, step, computation_time)
+        sleep(computation_time)
+        #tf.function(self.model)(epoch_number, step, computation_time)
 
     @dlp.log
     def get_loader(self, dataset_type=DatasetType.TRAIN):