Merge pull request #221 from NREL/gb/save_def_device

save default device to model parameters so model weights can be loade…
NREL · Jul 1, 2024 · ee6ab7b · ee6ab7b
2 parents a5f582f + b25afd5
commit ee6ab7b
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sup3r/models/base.py b/sup3r/models/base.py
@@ -351,6 +351,7 @@ def model_params(self):
             'means': means,
             'stdevs': stdevs,
             'meta': self.meta,
+            'default_device': self.default_device,
         }
 
         return model_params