Skip to content

Commit

Permalink
Fix (brevitas_examples/bnn_pynq): fix for no cuda error
Browse files Browse the repository at this point in the history
* code: Fixed `gpus` default to `None`

* readme: Fixed train and eval commands
  • Loading branch information
saadulkh authored Nov 30, 2023
1 parent d3219f3 commit 652ca8e
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions src/brevitas_examples/bnn_pynq/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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
```
2 changes: 1 addition & 1 deletion src/brevitas_examples/bnn_pynq/bnn_pynq_train.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def parse_args(args):
add_bool_arg(parser, "detect_nan", default=False)
# Compute resources
parser.add_argument("--num_workers", default=4, type=int, help="Number of workers")
parser.add_argument("--gpus", type=none_or_str, default="0", help="Comma separated GPUs")
parser.add_argument("--gpus", type=none_or_str, default=None, help="Comma separated GPUs")
# Optimizer hyperparams
parser.add_argument("--batch_size", default=100, type=int, help="batch size")
parser.add_argument("--lr", default=0.02, type=float, help="Learning rate")
Expand Down

0 comments on commit 652ca8e

Please sign in to comment.