Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
liunian-harold-li authored Apr 15, 2022
1 parent 9018c0e commit 8062244
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ python -m torch.distributed.launch --nnodes 2 --nproc_per_node=16 tools/train_ne
--skip-test --use-tensorboard --override_output_dir {output_dir}
```

For training GLIP-T models, we used `nnodes = 2`, `nproc_per_node=16` on 32GB A100 machines. For training GLIP-L models, we used `nnodes = 4`, `nproc_per_node=16` on 32GB A100 machines. Please setup the environment accordingly based on your local machine.
For training GLIP-T models, we used `nnodes = 2`, `nproc_per_node=16` on 32GB V100 machines. For training GLIP-L models, we used `nnodes = 4`, `nproc_per_node=16` on 32GB V100 machines. Please setup the environment accordingly based on your local machine.


## (Zero-Shot) Evaluation
Expand Down

0 comments on commit 8062244

Please sign in to comment.