Skip to content

Commit c872a52

Browse files
authored
[data][train] Fix compute config for local_fs + jpeg image classification benchmark (#52555)
Fix the misconfigured compute config introduced by #52453 Signed-off-by: Hao Chen <chenh1024@gmail.com>
1 parent 9928903 commit c872a52

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

release/release_tests.yaml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2160,8 +2160,6 @@
21602160
run:
21612161
timeout: 1200
21622162
script: RAY_TRAIN_V2_ENABLED=1 python train_benchmark.py --task=image_classification_jpeg --dataloader_type=torch --num_workers=16 --num_torch_workers=16 --skip_train_step --skip_validation_at_epoch_end
2163-
cluster:
2164-
cluster_compute: compute_configs/compute_gpu_1x1_aws.yaml
21652163

21662164
- __suffix__: skip_training.jpeg.local_fs
21672165
run:
@@ -2174,6 +2172,8 @@
21742172
run:
21752173
timeout: 1200
21762174
script: bash image_classification/localfs_image_classification_jpeg/download_input_data_from_s3.sh && RAY_TRAIN_V2_ENABLED=1 python train_benchmark.py --task=localfs_image_classification_jpeg --dataloader_type=torch --num_workers=1 --num_torch_workers=32 --skip_train_step --skip_validation_at_epoch_end
2175+
cluster:
2176+
cluster_compute: compute_configs/compute_gpu_1x1_aws.yaml
21772177

21782178

21792179
- name: train_horovod_multi_node_test

0 commit comments

Comments
 (0)