diff --git a/examples/power/conf/data/default.yaml b/examples/power/conf/data/default.yaml new file mode 100644 index 00000000..5eb78278 --- /dev/null +++ b/examples/power/conf/data/default.yaml @@ -0,0 +1,2 @@ +defaults: + - torch_mnist diff --git a/examples/power/conf/data/torch_mnist.yaml b/examples/power/conf/data/torch_mnist.yaml index 56e9c38b..48c63bbf 100644 --- a/examples/power/conf/data/torch_mnist.yaml +++ b/examples/power/conf/data/torch_mnist.yaml @@ -5,6 +5,8 @@ generate: sample: _target_: deckard.base.data.sampler.SklearnDataSampler random_state : 0 + train_size: 100 + test_size: 100 stratify: True sklearn_pipeline: _target_: deckard.base.data.sklearn_pipeline.SklearnDataPipeline diff --git a/examples/power/conf/deploy-cpu/cifar100.yaml b/examples/power/conf/deploy-cpu/cifar100.yaml index d335db02..227e5a5c 100644 --- a/examples/power/conf/deploy-cpu/cifar100.yaml +++ b/examples/power/conf/deploy-cpu/cifar100.yaml @@ -8,7 +8,7 @@ spec: image: ghcr.io/simplymathematics/deckard:main imagePullPolicy: Always workingDir: /deckard/examples/power/ - args: ["python", "-m", "deckard", "--config_file", "cifa100.yaml"] + args: ["python", "-m", "deckard", "--config_file", "cifar100.yaml"] env: - name: DECKARD_DEVICE value: "cpu" diff --git a/examples/power/conf/deploy-nvidia-tesla-a100/cifar100.yaml b/examples/power/conf/deploy-nvidia-tesla-a100/cifar100.yaml index fa13a34a..708506fc 100644 --- a/examples/power/conf/deploy-nvidia-tesla-a100/cifar100.yaml +++ b/examples/power/conf/deploy-nvidia-tesla-a100/cifar100.yaml @@ -8,7 +8,7 @@ spec: image: ghcr.io/simplymathematics/deckard:main imagePullPolicy: Always workingDir: /deckard/examples/power/ - args: ["python", "-m", "deckard", "--config_file", "cifa100.yaml"] + args: ["python", "-m", "deckard", "--config_file", "cifar100.yaml"] env: - name: DECKARD_DEVICE value: nvidia-tesla-a100 diff --git a/examples/power/conf/deploy-nvidia-tesla-p100/cifar100.yaml b/examples/power/conf/deploy-nvidia-tesla-p100/cifar100.yaml index fb57be6c..508503c5 100644 --- a/examples/power/conf/deploy-nvidia-tesla-p100/cifar100.yaml +++ b/examples/power/conf/deploy-nvidia-tesla-p100/cifar100.yaml @@ -8,7 +8,7 @@ spec: image: ghcr.io/simplymathematics/deckard:main imagePullPolicy: Always workingDir: /deckard/examples/power/ - args: ["python", "-m", "deckard", "--config_file", "cifa100.yaml"] + args: ["python", "-m", "deckard", "--config_file", "cifar100.yaml"] env: - name: DECKARD_DEVICE value: nvidia-tesla-p100 diff --git a/examples/power/conf/deploy-nvidia-tesla-v100/cifar100.yaml b/examples/power/conf/deploy-nvidia-tesla-v100/cifar100.yaml index d3e727e6..69fa0f92 100644 --- a/examples/power/conf/deploy-nvidia-tesla-v100/cifar100.yaml +++ b/examples/power/conf/deploy-nvidia-tesla-v100/cifar100.yaml @@ -8,7 +8,7 @@ spec: image: ghcr.io/simplymathematics/deckard:main imagePullPolicy: Always workingDir: /deckard/examples/power/ - args: ["python", "-m", "deckard", "--config_file", "cifa100.yaml"] + args: ["python", "-m", "deckard", "--config_file", "cifar100.yaml"] env: - name: DECKARD_DEVICE value: nvidia-tesla-v100