Wei 7ea6ccc2b3 Enable torchdynamo with torch_tensorrt(fx path) (#17765) ...

* enable fx2trt

* Update perf_train_gpu_one.mdx

* Update perf_train_gpu_one.mdx

* add lib check

* update

* format

* update

* fix import check

* fix isort

* improve doc

* refactor ctx manager

* fix isort

* black format

* isort fix

* fix format

* update args

* update black

* cleanups

* Update perf_train_gpu_one.mdx

* code refactor

* code refactor to init

* remove redundancy

* isort

* replace self.args with args

Co-authored-by: Stas Bekman <stas@stason.org>

2022-07-13 12:43:28 -04:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_data_collator.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_trainer_callback.py

Make predict() close progress bars after finishing (#17952) (#18078)

2022-07-08 16:44:24 -04:00

test_trainer_distributed.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_trainer_seq2seq.py

Reorganize file utils (#16264)

2022-03-23 10:26:33 -04:00

test_trainer_tpu.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_trainer_utils.py

Ensure tensors are at least 1d for pad and concat (#17179)

2022-05-11 13:19:08 -04:00

test_trainer.py

Enable torchdynamo with torch_tensorrt(fx path) (#17765)

2022-07-13 12:43:28 -04:00