Skip to content

Commit

Permalink
Merge pull request #380 from Miocio-nora/patch-1
Browse files Browse the repository at this point in the history
Update readme.md
  • Loading branch information
VainF committed Jun 6, 2024
2 parents ffa373c + 97f47f2 commit 7a324b5
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarks/readme.md
Original file line number Diff line number Diff line change
Expand Up @@ -102,7 +102,7 @@ wget https://github.com/VainF/Torch-Pruning/releases/download/v1.1.4/cifar100_vg
```
or train a new model:
```python
python main.py --mode pretrain --dataset vgg19 --model resnet56 --lr 0.1 --total-epochs 200 --lr-decay-milestones 120,150,180
python main.py --mode pretrain --dataset cifar100 --model vgg19 --lr 0.1 --total-epochs 200 --lr-decay-milestones 120,150,180
```

### 2.2 CIFAR-100 Pruning
Expand Down

0 comments on commit 7a324b5

Please sign in to comment.