Adrian Wälchli
9031dc3b81
Fix result gathering with varying tensor shapes ( #3020 )
...
* test for gethering results
* fix gather
* document tests
* changelog
* assert dtype
* default to concat
* additional test
2020-08-18 20:27:48 -04:00
Nathan Raw
b9695237f1
Save test predictions on multiple GPUs ( #2926 )
...
* Save test predictions on multiple GPUs
2020-08-14 17:52:43 -04:00
William Falcon
d13e5c9e53
document lightiningmodule better ( #2920 )
...
* updated docs
2020-08-11 19:39:43 -04:00
Justus Schock
fe29c53ab5
add ddp sync for logging in result step ( #2822 )
...
* add ddp sync for logging in result step
* pep8
* pep8
* make ddp tests run also on cpu (except windowws)
* create class instance in ddp test
* revert automated formatting
* pep8
2020-08-05 20:42:09 -04:00