diff --git a/NLQ/2D-TAN/README.md b/NLQ/2D-TAN/README.md index 3e10d44..491c7a4 100755 --- a/NLQ/2D-TAN/README.md +++ b/NLQ/2D-TAN/README.md @@ -10,10 +10,16 @@ pip install transformers ### Train && Test +To train a new model ``` python moment_localization/train.py --cfg experiments/ego4d/2D-TAN-40x40-K9L4-pool-window-std-sf.yaml --verbose ``` +To generate the submission file, do +``` +python moment_localization/test.py --cfg experiments/ego4d/2D-TAN-40x40-K9L4-pool-window-std-sf.yaml --verbose --split test --result nlq_2dtan_submission.json +``` + ## Acknowledgements This codebase is built on [2D-TAN](https://github.com/microsoft/2D-TAN).