Skip to content

Commit

Permalink
Documentation update
Browse files Browse the repository at this point in the history
  • Loading branch information
oguzhanbsolak committed Jan 10, 2024
1 parent a1f07ae commit 3016000
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 1 deletion.
8 changes: 7 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
# ADI MAX78000/MAX78002 Model Training and Synthesis

December 22, 2023
January 10, 2024

ADI’s MAX78000/MAX78002 project is comprised of five repositories:

Expand Down Expand Up @@ -1534,6 +1534,11 @@ The following table describes the most important command line arguments for `tra
| `--nas` | Enable network architecture search | |
| `--nas-policy` | Define NAS policy in YAML file | `--nas-policy nas/nas_policy.yaml` |
| `--regression` | Select regression instead of classification (changes Loss function, and log output) | |
| `--dr` | Set target embedding dimensionality for dimensionality reduction |`--dr 64` |
| `--scaf-lr` | Initial learning rate for sub-center ArcFace loss optimizer | |
| `--scaf-scale` |Scale hyperparameter for sub-center ArcFace loss | |
| `--scaf-margin` |Margin hyperparameter for sub-center ArcFace loss | |
| `--backbone-checkpoint` |Path to checkpoint from which to load backbone weights | |
| *Display and statistics* | | |
| `--enable-tensorboard` | Enable logging to TensorBoard (default: disabled) | |
| `--confusion` | Display the confusion matrix | |
Expand All @@ -1552,6 +1557,7 @@ The following table describes the most important command line arguments for `tra
| `--summary onnx_simplified` | Export trained model to simplified [ONNX](https://onnx.ai/) file (default name: model.onnx) | |
| `--summary-filename` | Change the file name for the exported model | `--summary-filename mnist.onnx` |
| `--save-sample` | Save data[index] from the test set to a NumPy pickle for use as sample data | `--save-sample 10` |
| `--slice-sample` | For models that require RGB input, when the sample from the dataset has additional channels, slice the sample into 3 channels | |
#### ONNX Model Export
Expand Down
Binary file modified README.pdf
Binary file not shown.

0 comments on commit 3016000

Please sign in to comment.