diff options
Diffstat (limited to 'src/training/experiments/sample_experiment.yml')
| -rw-r--r-- | src/training/experiments/sample_experiment.yml | 4 |
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 |