Merge pull request #574 from GiuseTripodi/fix_model_summary_reset_device_#542

rwood-97 · web-flow · commit a849a3acfb81 · 2026-01-05T08:53:56.000Z
Fix: Preserve model device in model_summary method
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -14,6 +14,10 @@ The following table shows which versions of MapReader are compatible with which
 ## Pre-release
 _Add new changes here_
 
+### Fixed
+
+- Fixes the `model_summary` method in the `ClassifierContainer` class ([#574](https://github.com/maps-as-data/MapReader/pull/574))
+
 ## [v1.8.2](https://github.com/Living-with-machines/MapReader/releases/tag/v1.8.2) (2025-12-19)
 
 ### Added
diff --git a/mapreader/classify/classifier.py b/mapreader/classify/classifier.py
@@ -565,7 +565,7 @@ def model_summary(
             col_names = ["output_size", "output_size", "num_params"]
 
         model_summary = summary(
-            self.model, input_size=input_size, col_names=col_names, **kwargs
+            self.model, input_size=input_size, col_names=col_names, device=self.device, **kwargs
         )
         print(model_summary)
 

Original file line number	Diff line number	Diff line change
`@@ -565,7 +565,7 @@ def model_summary(`
`565`	`565`	`col_names = ["output_size", "output_size", "num_params"]`
`566`	`566`
`567`	`567`	`model_summary = summary(`
`568`		`- self.model, input_size=input_size, col_names=col_names, **kwargs`
	`568`	`+ self.model, input_size=input_size, col_names=col_names, device=self.device, **kwargs`
`569`	`569`	`)`
`570`	`570`	`print(model_summary)`
`571`	`571`