Lezwon Castelino d9ea25590e fix ONNX model save on GPU (#3145) ... * added to(device) * added test * fix test on gpu * Update pytorch_lightning/core/lightning.py Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> * Update pytorch_lightning/core/lightning.py Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> * remove multi gpu check Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> * updated message * Update pytorch_lightning/core/lightning.py Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> * updated test * onxx to onnx * Update pytorch_lightning/core/lightning.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * Update tests/models/test_onnx.py Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * add no grad Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> * add isinstance back * chlog * error is input_sample is not Tensor Co-authored-by: Justus Schock <12886177+justusschock@users.noreply.github.com> Co-authored-by: Rohit Gupta <rohitgr1998@gmail.com> Co-authored-by: Jirka Borovec <jirka@pytorchlightning.ai>		2020-08-26 16:22:19 +00:00
..
__init__.py	document lightiningmodule better (#2920)	2020-08-11 19:39:43 -04:00
datamodule.py	Add transfer_batch_to_device hook to DataModule (#3038)	2020-08-20 08:47:11 -04:00
decorators.py	added copyright notices (#3062)	2020-08-19 22:03:22 -04:00
grads.py	added copyright notices (#3062)	2020-08-19 22:03:22 -04:00
hooks.py	remove on_perf check hooks (#3178)	2020-08-25 22:57:23 -04:00
lightning.py	fix ONNX model save on GPU (#3145)	2020-08-26 16:22:19 +00:00
memory.py	flake8 fixes (#3064)	2020-08-20 07:45:22 -04:00
saving.py	Make sure all kwargs come after args in _load_model_state() (#3063)	2020-08-20 07:29:44 -04:00
step_result.py	New modular metric interface (#2528)	2020-08-26 13:01:29 +02:00