diff --git a/README.md b/README.md index 82d1d620..8a3161e6 100644 --- a/README.md +++ b/README.md @@ -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: @@ -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 | | @@ -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 diff --git a/README.pdf b/README.pdf index 022f64b9..fbea4318 100644 Binary files a/README.pdf and b/README.pdf differ