Skip to content

Commit c185ec5

Browse files
committed
add
1 parent 92049b1 commit c185ec5

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

README.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,6 @@ Note that the result of DenseNet(k=12, T=36) is reported by original paper. The
137137
- An example to train a model on ImageNet:
138138

139139
```bash
140-
python train_imagenet.py --gpu [gpu ids (default: '0,1,2,3')] --workers [number of data loading workers (default: 8)] --epochs [number of total epochs to run (default: 100)] --start-epoch [manual epoch number (useful on restarts)] --batch-size [mini-batch size (default: 200)] --learning-rate [initial learning rate (default: 0.1)] --momentum [momentum (default: 0.9)] --weight-decay [weight decay (default: 1e-4)] --resume [path to latest checkpoint model (default: './cliquenet_s0.pth.tar')] --save [path to save models] --evaluate(evaluate model on validation set) --pretrained(use pre-trained model) [data:path to the imagenet dataset] ]
140+
python train_imagenet.py --gpu [gpu ids (default: '0,1,2,3')] --workers [number of data loading workers (default: 8)] --epochs [number of total epochs to run (default: 100)] --start-epoch [manual epoch number (useful on restarts)] --batch-size [mini-batch size (default: 200)] --learning-rate [initial learning rate (default: 0.1)] --momentum [momentum (default: 0.9)] --weight-decay [weight decay (default: 1e-4)] --resume [path to latest checkpoint model (default: './cliquenet_s0.pth.tar')] --save [path to save models] --evaluate(evaluate model on validation set) --pretrained(use pre-trained model) [data:path to the imagenet dataset]
141141
```
142142

0 commit comments

Comments
 (0)