readme: Fixed train and eval commands

Xilinx · Oct 28, 2023 · ac3a016 · ac3a016
1 parent a454a33
commit ac3a016
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/brevitas_examples/bnn_pynq/README.md b/src/brevitas_examples/bnn_pynq/README.md
@@ -31,17 +31,17 @@ A few notes on training:
 
 To start training a model from scratch, e.g. LFC_1W1A, run:
  ```bash
-BREVITAS_JIT=1 brevitas_bnn_pynq_train --network LFC_1W1A --experiments /path/to/experiments
+BREVITAS_JIT=1 python bnn_pynq_train.py --network LFC_1W1A --experiments /path/to/experiments
  ```
 
 ## Evaluate
 
 To evaluate a pretrained model, e.g. LFC_1W1A, run:
  ```bash
-BREVITAS_JIT=1 brevitas_bnn_pynq_train --evaluate --network LFC_1W1A --pretrained
+BREVITAS_JIT=1 python bnn_pynq_train.py --evaluate --network LFC_1W1A --pretrained
  ```
 
 To evaluate your own checkpoint, of e.g. LFC_1W1A, run:
  ```bash
-BREVITAS_JIT=1 brevitas_bnn_pynq_train --evaluate --network LFC_1W1A --resume /path/to/checkpoint.tar
+BREVITAS_JIT=1 python bnn_pynq_train.py --evaluate --network LFC_1W1A --resume /path/to/checkpoint.tar
  ```