Changing the number of epoch based on previous observation. (#1222)

GoogleCloudPlatform · Jul 12, 2023 · 8cf5254 · 8cf5254
1 parent 4f04191
commit 8cf5254
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/perfmetrics/scripts/ml_tests/pytorch/dino/setup_container.sh b/perfmetrics/scripts/ml_tests/pytorch/dino/setup_container.sh
@@ -74,7 +74,7 @@ gsutil cp start_time.txt $ARTIFACTS_BUCKET_PATH/
     --norm_last_layer False \
     --use_fp16 False \
     --clip_grad 0 \
-    --epochs 100 \
+    --epochs 80 \
     --global_crops_scale 0.25 1.0 \
     --local_crops_number 10 \
     --local_crops_scale 0.05 0.25 \

diff --git a/perfmetrics/scripts/ml_tests/tf/resnet/setup_scripts/resnet_runner.py b/perfmetrics/scripts/ml_tests/tf/resnet/setup_scripts/resnet_runner.py
@@ -111,5 +111,5 @@
     params=exp_config,
     model_dir=model_dir,
     run_post_eval=True,
-    epochs=1000,
+    epochs=675,
     clear_kernel_cache=True)