summaryrefslogtreecommitdiff
path: root/src/training/experiments/sample_experiment.yml
diff options
context:
space:
mode:
Diffstat (limited to 'src/training/experiments/sample_experiment.yml')
-rw-r--r--src/training/experiments/sample_experiment.yml4
1 files changed, 2 insertions, 2 deletions
diff --git a/src/training/experiments/sample_experiment.yml b/src/training/experiments/sample_experiment.yml
index bae02ac..b00bd5a 100644
--- a/src/training/experiments/sample_experiment.yml
+++ b/src/training/experiments/sample_experiment.yml
@@ -26,8 +26,8 @@ experiments:
network_args:
in_channels: 1
num_classes: 80
- depths: [1, 1]
- block_sizes: [128, 256]
+ depths: [2, 1]
+ block_sizes: [96, 32]
# network: LeNet
# network_args:
# output_size: 62