lightning/tests/models
ananthsub f6d892ac21
[feat] Support custom filesystems in LightningModule.to_torchscript (#7617)
* [feat] Support custom filesystems in LightningModule.to_torchscript

* Update CHANGELOG.md

* Update test_torchscript.py

* Update test_torchscript.py

* Update CHANGELOG.md

* Update test_torchscript.py
2021-05-21 11:23:15 +00:00
..
conf bugfix: Resolve interpolation bug with Hydra (#5406) 2021-01-26 14:29:47 +01:00
data `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_cpu.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_gpu.py make gpus=str in Trainer consistent with command line parsing of string (#6388) 2021-05-04 09:56:27 +00:00
test_grad_norm.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_hooks.py Add `dataloader_idx` to batch transfer hooks (#6241) 2021-05-13 23:03:55 +05:30
test_horovod.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_hparams.py fix save_hyperparameters(container) if container is empty (#7268) 2021-04-30 13:38:42 +01:00
test_onnx.py Refactor: skipif for Windows 2/n (#6268) 2021-03-02 09:36:01 +00:00
test_restore.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_sync_batchnorm.py `TrainerState` refactor [5/5] (#7173) 2021-05-04 12:50:56 +02:00
test_torchscript.py [feat] Support custom filesystems in LightningModule.to_torchscript (#7617) 2021-05-21 11:23:15 +00:00
test_tpu.py Use `torch.nn.utils.clip_grad_norm_` and add `clip_grad_by_value` support for TPU (#7025) 2021-05-07 16:41:39 +00:00