Skip to content

Commit 0096534

Browse files
committed
docs: update ResNet-finetune README with new CLI example for activation replacement and cautious weight decay
- Added command showcasing `--replace-activation` and `--cautious-weight-decay` usage.
1 parent 6587eb8 commit 0096534

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

examples/resnet_finetune/README.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,7 @@ Args {
117117
weight_decay: 0.02,
118118
}
119119
120+
$ cargo run --release -p resnet_finetune --features cuda -- --replace-activation leaky-relu --cautious-weight-decay
120121
| Split | Metric | Min. | Epoch | Max. | Epoch |
121122
|-------|--------------------------------|----------|----------|----------|----------|
122123
| Train | CPU Memory | 16.486 | 1 | 18.592 | 60 |

0 commit comments

Comments
 (0)