From 1937d2eff549356c996033b3f1335ffa87a0d527 Mon Sep 17 00:00:00 2001 From: Karan Jariwala Date: Wed, 21 Sep 2022 13:51:06 -0700 Subject: [PATCH] Removed streaming version 2 from yaml since version 1 is default (#1551) Co-authored-by: bandish-shah <86627118+bandish-shah@users.noreply.github.com> --- composer/yamls/models/bert-base.yaml | 2 -- composer/yamls/models/deeplabv3_streaming_ade20k_optimized.yaml | 2 -- composer/yamls/models/resnet50_streaming.yaml | 2 -- composer/yamls/models/resnet56_streaming_cifar10.yaml | 2 -- 4 files changed, 8 deletions(-) diff --git a/composer/yamls/models/bert-base.yaml b/composer/yamls/models/bert-base.yaml index c351edde23..93a1a0251c 100644 --- a/composer/yamls/models/bert-base.yaml +++ b/composer/yamls/models/bert-base.yaml @@ -8,7 +8,6 @@ model: # Train the model on the English C4 corpus train_dataset: streaming_c4: - version: 2 remote: s3://mosaicml-internal-dataset-c4/mds/2/ local: /tmp/mds-cache/mds-c4/ split: train @@ -32,7 +31,6 @@ evaluators: - label: bert_pre_training eval_dataset: streaming_c4: - version: 2 remote: s3://mosaicml-internal-dataset-c4/mds/2/ local: /tmp/mds-cache/mds-c4/ split: val diff --git a/composer/yamls/models/deeplabv3_streaming_ade20k_optimized.yaml b/composer/yamls/models/deeplabv3_streaming_ade20k_optimized.yaml index b2bf6badf4..ea63595fa7 100644 --- a/composer/yamls/models/deeplabv3_streaming_ade20k_optimized.yaml +++ b/composer/yamls/models/deeplabv3_streaming_ade20k_optimized.yaml @@ -1,6 +1,5 @@ train_dataset: streaming_ade20k: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-ade20k/ split: train @@ -13,7 +12,6 @@ train_dataset: shuffle: true val_dataset: streaming_ade20k: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-ade20k/ split: val diff --git a/composer/yamls/models/resnet50_streaming.yaml b/composer/yamls/models/resnet50_streaming.yaml index e661f5cd5c..cbca1dd72d 100644 --- a/composer/yamls/models/resnet50_streaming.yaml +++ b/composer/yamls/models/resnet50_streaming.yaml @@ -1,6 +1,5 @@ train_dataset: streaming_imagenet1k: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-imagenet1k/ split: train @@ -10,7 +9,6 @@ train_dataset: drop_last: true val_dataset: streaming_imagenet1k: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-imagenet1k/ split: val diff --git a/composer/yamls/models/resnet56_streaming_cifar10.yaml b/composer/yamls/models/resnet56_streaming_cifar10.yaml index b6ff388cb9..17de3f0b43 100644 --- a/composer/yamls/models/resnet56_streaming_cifar10.yaml +++ b/composer/yamls/models/resnet56_streaming_cifar10.yaml @@ -1,6 +1,5 @@ train_dataset: streaming_cifar10: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-cifar10/ split: train @@ -8,7 +7,6 @@ train_dataset: drop_last: true val_dataset: streaming_cifar10: - version: 2 remote: your-bucket-here local: /tmp/mds-cache/mds-cifar10/ split: val