aboutsummaryrefslogtreecommitdiff
path: root/src/mlia/resources/optimization_profiles/optimization-conv2d-pruning.toml
blob: 87822ed52aa545833efa65347c148f7f6cf5bc7c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# 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