test on KS

.github/workflows/cnn_e2e.yml

@@ -152,7 +152,7 @@ jobs:
     if: always()
     runs-on: ['self-hosted', 'linux', 'x64', 'gpu-8-titan-v']
     env:
-      TEST_CONTAINER_NAME: "oneflow_benchmark-run-id-${{ github.run_id }}-${{ matrix.entry }}-test"
+      TEST_CONTAINER_NAME: "oneflow_benchmark-run-id-${{ github.run_id }}-${{ github.event.inputs.compute_platform }}-test"
       TEST_WITH_TORCH_IMG_TAG: registry.cn-beijing.aliyuncs.com/oneflow/test-with-pytorch-1.9.0:e7a497b41d8b7f1bce055b1f23d027f93b1557ae
     steps:
       - name: Fix permissions

ci/test/resnet50_e2e.sh

@@ -31,7 +31,7 @@ python3 ${E2E_SRC_ROOT}/of_cnn_train_val.py \
      --nccl_fusion_max_ops=24 \
      --gpu_image_decoder=True \
      --num_epoch=$E2E_NUM_EPOCHS \
-     --num_examples=1024 \
+     --num_examples=1281167 \
      --model=${model} 2>&1 | tee ${LOGFILE}
 
 echo "Writting log to ${LOGFILE}"