diff options
author | Nathan Bailey <nathan.bailey@arm.com> | 2024-04-15 15:34:03 +0100 |
---|---|---|
committer | Nathan Bailey <nathan.bailey@arm.com> | 2024-05-08 12:15:48 +0000 |
commit | 198ba5eed95677ddb9d1e8c4119062dd3412510a (patch) | |
tree | c37154ab0a30a0c76c8ad275aae8ee4e40e2a1b2 /src/mlia/resources/optimization_profiles/optimization.toml | |
parent | 0999ba0f4381ce1e2e8b06a932bfe693692223e2 (diff) | |
download | mlia-198ba5eed95677ddb9d1e8c4119062dd3412510a.tar.gz |
feat: Enables augmentations via --optimization-profile
- Enables user to specify augmentations via the --optimization-profile switch
- Can specify from pre-given examples or can provide each parameter manually
- Updates README
Resolves: MLIA-1147
Signed-off-by: Nathan Bailey <nathan.bailey@arm.com>
Change-Id: I9cbe71d85def6a8db9dc974adc4bcc8d90625505
Diffstat (limited to 'src/mlia/resources/optimization_profiles/optimization.toml')
-rw-r--r-- | src/mlia/resources/optimization_profiles/optimization.toml | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/src/mlia/resources/optimization_profiles/optimization.toml b/src/mlia/resources/optimization_profiles/optimization.toml index 623a763..42b64f0 100644 --- a/src/mlia/resources/optimization_profiles/optimization.toml +++ b/src/mlia/resources/optimization_profiles/optimization.toml @@ -7,5 +7,6 @@ learning_rate = 1e-3 show_progress = true steps = 48000 learning_rate_schedule = "cosine" +augmentations = "gaussian" num_procs = 1 num_threads = 0 |