lightning/pytorch_lightning/core
Lezwon Castelino 3910ad0330
bugfix/3185 transpose (#3252)
* change t() to transpose() as xla devices do not support .t() on 1-dim tensor

* detach tensor before copying

* Revert "detach tensor before copying"

This reverts commit 37cc7bbe

* changed dims

* added test_result_obj_on_tpu

* detach before copying

* detach before copying

* detach before copying

* replace torch.cat with sum
2020-09-01 09:17:52 -04:00
..
__init__.py document lightiningmodule better (#2920) 2020-08-11 19:39:43 -04:00
datamodule.py Parse Union[bool, str] arguments (#3235) 2020-08-29 10:39:42 -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 Bug Fix: Remote Logging with Tensorboard (#3236) 2020-08-28 07:01:35 -04:00
step_result.py bugfix/3185 transpose (#3252) 2020-09-01 09:17:52 -04:00