blob: aa7f982c7572738e0b3a117f28013e527c4c4bb5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
# SPDX-FileCopyrightText: Copyright 2024, Arm Limited and/or its affiliates.
# SPDX-License-Identifier: Apache-2.0
[rewrite.training_parameters]
batch_size = 32
learning_rate = 1e-3
show_progress = true
steps = 48000
learning_rate_schedule = "cosine"
num_procs = 1
num_threads = 0
augmentations.gaussian_strength = 0.0
augmentations.mixup_strength = 0.0
[rewrite.conv2d-sparsity]
sparsity_m = 2
sparsity_n = 4
activation = "relu"
kernel_size = [3, 3]
|
