diff --git a/torchbenchmark/models/ADDING_MODELS.md b/torchbenchmark/models/ADDING_MODELS.md index 8fdb646a38..b993f663f3 100644 --- a/torchbenchmark/models/ADDING_MODELS.md +++ b/torchbenchmark/models/ADDING_MODELS.md @@ -89,3 +89,13 @@ make sure to verify correctness by capturing the output of the non-jit model for version. [PyTorch Docs: Creating Torchscript Code](https://pytorch.org/docs/1.1.0/jit.html#creating-torchscript-code) + +### Test + +After you've submitted your new model, suppose it was called `new_model` make sure the tests pass locally. Your model name is equivalent to the new folder you'd have created in `torchbenchmark/models` + +1. `cd benchmark` +2. `python install.py` +3. `python test.py -k "test_new_model"` + +And thank you for contributing to torchbench!