Skip to content

Commit

Permalink
Test only on CPU
Browse files Browse the repository at this point in the history
  • Loading branch information
bryanbocao authored Jun 24, 2022
1 parent 22676b4 commit 16eb6f8
Showing 1 changed file with 12 additions and 6 deletions.
18 changes: 12 additions & 6 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,16 +9,22 @@ I'm playing with [PyTorch](http://pytorch.org/) on the CIFAR10 dataset.
## Training
```
# Start training with:
python main.py --net VGG16 --train True --test True
python main.py --net ResNet18 --train --test
# You can manually resume the training with:
python main.py --net VGG16 --train True --test True --resume --lr=0.01
python main.py --net ResNet18 --train --test --resume --lr=0.01
```

## Testing
```
# Test only on GPU
python main.py --net ResNet18 --test
# Test only
python main.py --net VGG16 --test True
# Test only on GPU with pruning (0.3)
python main.py --net ResNet18 --test --prune --pruning_rate 0.3
# Test only with pruning (0.3)
python main.py --net VGG16 --test True --prune True --pruning_rate 0.3
# Test only on CPU
python main.py --net ResNet18 --test --select_device cpu
```

## Accuracy
Expand Down

0 comments on commit 16eb6f8

Please sign in to comment.