mirror of
https://github.com/wassname/pytorch-lightning.git
synced 2026-09-02 12:40:56 +08:00
Adrian WälchliandJirka Borovec
8211256c46
* refactor and added hook variant a variant b add test revert rename add changelog docs * resolve merge duplication * overridden typo * fix test * tpu id * raise if TPU not available * re-use apply_to_collection function for parsing collections * comment * make utility function available to user * documentation * move changelog entry to top * fix tpu transfer call * fix call * remove hardcoded string * improve test * call model hook by default * Apply suggestions from code review * rename utility function Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
5 lines
179 B
Python
5 lines
179 B
Python
"""General utilities"""
|
|
|
|
from pytorch_lightning.utilities.distributed import rank_zero_only, rank_zero_warn
|
|
from pytorch_lightning.utilities.apply_func import move_data_to_device
|