thomas chaton 013756404b [bugfix] Add set_default_tensor_type to torch.DoubleTensor with precision=64 (#7108) ... * update * Update pytorch_lightning/plugins/precision/double.py Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * Update pytorch_lightning/plugins/precision/double.py Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * Update pytorch_lightning/plugins/precision/double.py Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com> * resolve tests Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>		2021-04-20 15:25:37 +00:00
..
__init__.py	Feature/double precision (#6595)	2021-03-24 15:47:58 +05:30
apex_amp.py	Create pytorch_lightning/utilities/types.py (#7048)	2021-04-19 14:43:16 +02:00
deepspeed_precision.py	Typing for accelerators and plugins (#7022)	2021-04-15 16:48:16 +00:00
double.py	[bugfix] Add set_default_tensor_type to torch.DoubleTensor with precision=64 (#7108)	2021-04-20 15:25:37 +00:00
mixed.py	Add mypy typing to precision plugins. (#6149)	2021-02-26 14:27:16 +01:00
native_amp.py	[bugfix] Add set_default_tensor_type to torch.DoubleTensor with precision=64 (#7108)	2021-04-20 15:25:37 +00:00
precision_plugin.py	Create pytorch_lightning/utilities/types.py (#7048)	2021-04-19 14:43:16 +02:00
sharded_native_amp.py	Typing for accelerators and plugins (#7022)	2021-04-15 16:48:16 +00:00
tpu_bfloat.py	Create pytorch_lightning/utilities/types.py (#7048)	2021-04-19 14:43:16 +02:00