diff --git a/flutter/cpp/flutter/dart_run_benchmark.cc b/flutter/cpp/flutter/dart_run_benchmark.cc
index 5be456540..27838005e 100644
--- a/flutter/cpp/flutter/dart_run_benchmark.cc
+++ b/flutter/cpp/flutter/dart_run_benchmark.cc
@@ -73,7 +73,7 @@ struct dart_ffi_run_benchmark_out* dart_ffi_run_benchmark(
     case ::mlperf::mobile::DatasetConfig::IMAGENET:
       dataset = std::make_unique<::mlperf::mobile::Imagenet>(
           backend.get(), in->dataset_data_path, in->dataset_groundtruth_path,
-          in->dataset_offset, 224, 224 /* width, height */);
+          in->dataset_offset, 256, 256 /* width, height */);
       break;
     case ::mlperf::mobile::DatasetConfig::COCO:
       dataset = std::make_unique<::mlperf::mobile::Coco>(
diff --git a/mobile_back_pixel/cpp/backend_tflite/settings/tflite_settings_pixel6.pbtxt b/mobile_back_pixel/cpp/backend_tflite/settings/tflite_settings_pixel6.pbtxt
index 439c0826b..c77d53696 100644
--- a/mobile_back_pixel/cpp/backend_tflite/settings/tflite_settings_pixel6.pbtxt
+++ b/mobile_back_pixel/cpp/backend_tflite/settings/tflite_settings_pixel6.pbtxt
@@ -35,7 +35,7 @@ benchmark_setting {
   accelerator: "tpu"
   accelerator_desc: "Google Edge TPU"
   framework: "TFLite NNAPI"
-  model_path: "local:///small.tflite"
+  model_path: "local:///medium.tflite"
   model_checksum: "60f651765fa0d52b468f2e1c15c47f91"
 }
 
@@ -67,7 +67,7 @@ benchmark_setting {
   accelerator_desc: "Google Edge TPU"
   framework: "TFLite NNAPI"
   batch_size: 16
-  model_path: "local:///small.tflite"
+  model_path: "local:///medium.tflite"
   model_checksum: "60f651765fa0d52b468f2e1c15c47f91"
 }
 
diff --git a/mobile_back_tflite/cpp/backend_tflite/backend_settings/tflite_settings_android.pbtxt b/mobile_back_tflite/cpp/backend_tflite/backend_settings/tflite_settings_android.pbtxt
index fa036f0f9..edff25915 100644
--- a/mobile_back_tflite/cpp/backend_tflite/backend_settings/tflite_settings_android.pbtxt
+++ b/mobile_back_tflite/cpp/backend_tflite/backend_settings/tflite_settings_android.pbtxt
@@ -38,7 +38,7 @@ benchmark_setting {
     delegate_name: "NNAPI"
     accelerator_name: "npu"
     accelerator_desc: "NPU"
-    model_path: "local:///small.tflite"
+    model_path: "local:///medium.tflite"
     model_checksum: "60f651765fa0d52b468f2e1c15c47f91"
   }
   delegate_choice: {
@@ -80,7 +80,7 @@ benchmark_setting {
     delegate_name: "NNAPI"
     accelerator_name: "npu"
     accelerator_desc: "NPU"
-    model_path: "local:///small.tflite"
+    model_path: "local:///medium.tflite"
     model_checksum: "60f651765fa0d52b468f2e1c15c47f91"
     batch_size: 2
   }