2023/09/04-15:35:13 (Linux sv2111 unknown)

pbenner · Sep 4, 2023 · 8e892e5 · 8e892e5
1 parent 8b24dc2
commit 8e892e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/coordinationnet/model.py b/coordinationnet/model.py
@@ -54,9 +54,9 @@ def __init__(self, cache_file = None, **kwargs):
 
         if self.lit_model.global_rank == 0:
 
-            print(f'{model_config}')
+            print(f'{self.lit_model.model.model_config}')
 
-            print(f'Creating a GNN model with {self.n_parameters:,} parameters')
+            print(f'Creating a GNN model with {self.lit_model.model.n_parameters:,} parameters')
 
     def fit_scaler(self, data : LitCoordinationFeaturesData):
         y = torch.cat([ y_batch for _, y_batch in data.get_dataloader(data.data) ])