diff --git a/dlio_benchmark/configs/workload/resnet50.yaml b/dlio_benchmark/configs/workload/resnet50.yaml
index 389524f9..5fa2ab71 100644
--- 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
index 2b81524c..4a319450 100644
--- 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):