Skip to content

Commit

Permalink
Add sweep config for cifar benchmark
Browse files Browse the repository at this point in the history
  • Loading branch information
mtpauly committed Apr 26, 2023
1 parent 124c106 commit 56db453
Showing 1 changed file with 48 additions and 0 deletions.
48 changes: 48 additions & 0 deletions sweep_configs/cifar_sequential_benchmark.yaml
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
name: cifar_no_dropout
description: Benchmark the network with no dropout
project: k-dropout
program: train_net.py
method: grid
metric:
name: test_acc
goal: minimize
parameters:
# experiment
seed:
values: [1, 2, 3, 4, 5]
# training
device:
value: cuda
epochs:
value: 300
lr:
value: 0.0005
# model
input_size:
value: 3072
hidden_size:
value: 2000
output_size:
value: 10
n_hidden:
value: 2
# dropout
dropout_layer:
value: none
input_p:
value: 0.
p:
value: 0.
k:
values: [1]
m:
values: [-1]
# dataset
dataset_name:
value: cifar10
batch_size:
value: 512
test_batch_size:
value: 512
num_workers:
value: 4

0 comments on commit 56db453

Please sign in to comment.