Skip to content

Commit

Permalink
[tools/onert_train] Add onert_train usage to readme (#12698)
Browse files Browse the repository at this point in the history
This PR adds onert_train usage to the readme.

ONE-DCO-1.0-Signed-off-by: SeungHui Youn <[email protected]>
  • Loading branch information
zetwhite authored Feb 28, 2024
1 parent aaaeb13 commit d003f92
Showing 1 changed file with 15 additions and 2 deletions.
17 changes: 15 additions & 2 deletions tests/tools/onert_train/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,19 @@ sudo apt install -y libhdf5-dev libboost-program-options-dev

## Usage

### Simple train
You could train your model using the command like below.

```bash
onert_train \
--path [circle file or nnpackage] \
--load_input:raw [training input data] \
--load_expected:raw [training output data] \
--batch_size 32 \
--epoch 5 \
--optimizer 1 \ # sgd
--learning_rate 0.01 \
--loss 2 \ # cateogrical crossentropy
--loss_reduction_type 1 # sum over batch size
```

### Simple inference to trained model
`onert_train --help` would help you to set each parameter.

0 comments on commit d003f92

Please sign in to comment.