Commit 5bf4cc7d authored by RangiLyu's avatar RangiLyu Committed by Facebook GitHub Bot
Browse files

Typo fixes

Summary:
Change depoyment to deployment in README.md.
Change datasest to datasets in tools/exporter.py.

Pull Request resolved: https://github.com/facebookresearch/d2go/pull/7

Reviewed By: newstzpz

Differential Revision: D26821039

Pulled By: zhanghang1989

fbshipit-source-id: 5056d15c877c4b3d771d33267139e73f1527da21
parent a8aaa5b1
......@@ -6,7 +6,7 @@ D2Go is a production ready software system from FacebookResearch, which supports
- It is a deep learning toolkit powered by [PyTorch](https://pytorch.org/) and [Detectron2](https://github.com/facebookresearch/detectron2).
- State-of-the-art efficient backbone networks for mobile devices.
- End-to-end model training, quantization and depoyment pipeline.
- End-to-end model training, quantization and deployment pipeline.
- Easy export to TorchScript format for deployment.
## Installation
......
......@@ -43,8 +43,8 @@ def main(
# NOTE: train dataset is used to avoid leakage since the data might be used for
# running calibration for quantization. test_loader is used to make sure it follows
# the inference behaviour (augmentation will not be applied).
datasest = cfg.DATASETS.TRAIN[0]
data_loader = runner.build_detection_test_loader(cfg, datasest)
datasets = cfg.DATASETS.TRAIN[0]
data_loader = runner.build_detection_test_loader(cfg, datasets)
logger.info("Running the pytorch model and print FLOPS ...")
first_batch = next(iter(data_loader))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment