Add DCP Saver module to the callbacks target (#775)

Summary: Pull Request resolved: #775

Reviewed By: JKSenthil

Differential Revision: D55618741

fbshipit-source-id: 46e8192ca8777450d111dd5570d13d2ccb2e7d2e

Author: diego-urgell

torchtnt/framework/callbacks/__init__.py

@@ -7,6 +7,7 @@
 # pyre-strict
 
 from .base_csv_writer import BaseCSVWriter
+from .dcp_saver import DistributedCheckpointSaver
 from .early_stopping import EarlyStopping
 from .empty_cuda_cache import EmptyCudaCache
 from .garbage_collector import GarbageCollector
@@ -44,4 +45,5 @@
     "TorchSnapshotSaver",
     "TQDMProgressBar",
     "TrainProgressMonitor",
+    "DistributedCheckpointSaver",
 ]