From f2ba0a4aa9949f6b30d60f4f7fef232021fabe4c Mon Sep 17 00:00:00 2001 From: Jeff Yang Date: Sat, 24 Oct 2020 19:10:47 +0630 Subject: [PATCH 1/2] Update setup.cfg --- setup.cfg | 2 ++ 1 file changed, 2 insertions(+) diff --git a/setup.cfg b/setup.cfg index 00608f0487c0..cae9c5153343 100644 --- a/setup.cfg +++ b/setup.cfg @@ -28,3 +28,5 @@ markers = distributed: mark a test with distributed option multinode_distributed: mark a test with multi-node distributed option tpu: mark a test as requiring XLA +addopts = + --color=yes From d563a81c025f5cf585561b5057947b9694010726 Mon Sep 17 00:00:00 2001 From: ydcjeff Date: Sat, 14 Nov 2020 15:03:07 +0630 Subject: [PATCH 2/2] [docs] missing function in handlers docs --- docs/source/handlers.rst | 2 ++ ignite/handlers/checkpoint.py | 2 +- 2 files changed, 3 insertions(+), 1 deletion(-) diff --git a/docs/source/handlers.rst b/docs/source/handlers.rst index d7392e546916..42545fefbfe8 100644 --- a/docs/source/handlers.rst +++ b/docs/source/handlers.rst @@ -26,3 +26,5 @@ Complete list of handlers :members: .. autoclass:: TerminateOnNan + +.. autofunction:: global_step_from_engine diff --git a/ignite/handlers/checkpoint.py b/ignite/handlers/checkpoint.py index 3090b8eac5f3..f3c419ee9dfb 100644 --- a/ignite/handlers/checkpoint.py +++ b/ignite/handlers/checkpoint.py @@ -66,7 +66,7 @@ def remove(self, filename: str) -> None: class Checkpoint(Serializable): """Checkpoint handler can be used to periodically save and load objects which have attribute - ``state_dict`/`load_state_dict``. This class can use specific save handlers to store on the disk or a cloud + ``state_dict/load_state_dict``. This class can use specific save handlers to store on the disk or a cloud storage, etc. The Checkpoint handler (if used with :class:`~ignite.handlers.DiskSaver`) also handles automatically moving data on TPU to CPU before writing the checkpoint.