transformers/tests/decision_transformer
Yih-Dar c04619ecf3
Enable more test_torchscript (#16679)
* update _create_and_check_torchscript

* Enable test_torchscript

* clear_class_registry

Co-authored-by: ydshieh <ydshieh@users.noreply.github.com>
2022-04-11 18:23:35 +02:00
..
__init__.py Decision transformer gym (#15845) 2022-03-23 16:18:43 -04:00
test_modeling_decision_transformer.py Enable more test_torchscript (#16679) 2022-04-11 18:23:35 +02:00