lightning/tests/models
Jean-Baptiste SCHIRATTI cea5f1f538
Fix for `load_from_checkpoint` (#2776)
* Fix.

* Fix #2550: allow to load model from checkpoint if self.save_hyperparameters() was not called.

* Fix? Cleaner way of not calling self.save_hyperparameters in EvalModelTemplate.

* Fix? `_load_model_state` cleanup

* Fix?

* Fix #2550: allow to load model from checkpoint if self.save_hyperparameters() was not called.

* Fix.

* Fix? Cleaner way of not calling self.save_hyperparameters in EvalModelTemplate.

* Fix? `_load_model_state` cleanup

* Fixed side effect in `test_load_model_from_checkpoint_extra_args`.

* Apply suggestions from code review

* fix

* try

* fixed missing arg in evalmodel

* fixed missing arg in evalmodel

* fix

* update

* fix loading

* add test

* prune

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>
Co-authored-by: William Falcon <waf2107@columbia.edu>
2020-10-05 12:44:23 -04:00
..
data/horovod part 5 of #3733 (#3774) 2020-10-01 12:34:12 -04:00
__init__.py CI: split tests-examples (#990) 2020-03-25 07:46:27 -04:00
test_amp.py ref: bug fix with logging val epoch end + monitor (#3812) 2020-10-03 12:33:29 -04:00
test_cpu.py [WIP] ref: deprecated results obj, added support for simpler comms (1/n) (#3681) 2020-09-27 23:19:46 -04:00
test_gpu.py Deprecate early_stop_callback Trainer argument (part 2) (#3845) 2020-10-04 17:36:47 -04:00
test_grad_norm.py fix gradient norm tracking for row_log_interval > 1 (#3489) 2020-09-15 18:41:27 +02:00
test_hooks.py Fix `on_train_batch_start` hook to end epoch early (#3700) 2020-10-02 21:46:46 +02:00
test_horovod.py Fix global step increment on training_epoch_end (#3673) 2020-09-27 20:19:51 -04:00
test_hparams.py Fix for `load_from_checkpoint` (#2776) 2020-10-05 12:44:23 -04:00
test_onnx.py fix ONNX model save on GPU (#3145) 2020-08-26 16:22:19 +00:00
test_restore.py Fix for `load_from_checkpoint` (#2776) 2020-10-05 12:44:23 -04:00
test_sync_batchnorm.py updated sync bn (#2838) 2020-08-06 01:12:11 +02:00
test_torchscript.py to_torchscript method for LightningModule (#3258) 2020-09-03 20:24:44 +02:00
test_tpu.py Deprecate early_stop_callback Trainer argument (part 2) (#3845) 2020-10-04 17:36:47 -04:00