UniversalAlgorithmic commited on
Commit
f83ae64
·
verified ·
1 Parent(s): 7213cec

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -0
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\