Update README.md
Browse files
README.md
CHANGED
|
@@ -429,6 +429,11 @@ For Network Architecture Search, please run the following command to evaluate o
|
|
| 429 |
|
| 430 |
cd neural-architecture-search
|
| 431 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 432 |
# Required: Download our ResNet-34 weights to /path/to/neural-architecture-search/resnet34
|
| 433 |
torchrun --nproc_per_node=4 train.py\
|
| 434 |
--data-path /home/cs/Documents/datasets/imagenet\
|
|
|
|
| 429 |
|
| 430 |
cd neural-architecture-search
|
| 431 |
|
| 432 |
+
# Required: Download our ResNet-18 weights to /path/to/neural-architecture-search/resnet18
|
| 433 |
+
torchrun --nproc_per_node=4 train.py\
|
| 434 |
+
--data-path /home/cs/Documents/datasets/imagenet\
|
| 435 |
+
--model resnet18 --resume resnet18/model_3.pth --test-only
|
| 436 |
+
|
| 437 |
# Required: Download our ResNet-34 weights to /path/to/neural-architecture-search/resnet34
|
| 438 |
torchrun --nproc_per_node=4 train.py\
|
| 439 |
--data-path /home/cs/Documents/datasets/imagenet\
|