Fixes #292 (#303)

* early stopping callback is not default * added a default logger * added default checkpoint callback * added default checkpoint/loggers * added default checkpoint/loggers * updated docs * cleaned demos * cleaned demos * cleaned demos * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers * clean up docs around loggers
2026-09-12 12:40:20 +08:00 · 2019-10-04 19:48:57 -04:00
parent a578de511d
commit bf09060fef
17 changed files with 146 additions and 201 deletions
@@ -35,7 +35,7 @@ class ModelSummary(object):
        out_sizes = []
        input_ = self.model.example_input_array

-        if self.model.use_ddp or self.model.use_dp:
+        if self.model.use_ddp or self.model.use_dp or self.model.single_gpu:
            input_ = input_.cuda(0)

        if self.model.trainer.use_amp: