Davout 0.1.1.dev165__tar.gz → 0.1.1.dev167__tar.gz

{davout-0.1.1.dev165/source/Davout.egg-info → davout-0.1.1.dev167}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Davout
-Version: 0.1.1.dev165
+Version: 0.1.1.dev167
 Summary: Scientific computing utilities combining FEM, ANN and tooling
 Author: Matheus Janczkowski
 Author-email: Matheus Janczkowski <matheusj2009@hotmail.com>

{davout-0.1.1.dev165 → davout-0.1.1.dev167}/pyproject.toml

@@ -63,7 +63,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "Davout"
-version = "0.1.1.dev165"
+version = "0.1.1.dev167"
 description = "Scientific computing utilities combining FEM, ANN and tooling"
 readme = "README.md"
 authors = [

{davout-0.1.1.dev165 → davout-0.1.1.dev167}/source/Davout/DeepMech/tool_box/training_tools.py

@@ -54,6 +54,10 @@ class ModelTraining:
 
         self.verbose = verbose
 
+        # Gets the number of trainable parameters of the model
+
+        self.number_trainable_parameters = self.model.count_params()
+
         # Saves the parent path where to save the model
 
         if parent_path=="get current path":
@@ -146,7 +150,11 @@ class ModelTraining:
 
                     print("Iteration "+integer_toString(i, max_digits)+
                     ": loss="+format(loss_value.numpy(), '.5e')+", gra"+
-                    "dient norm: "+format(gradient_norm.numpy(), '.5e'))
+                    "dient norm: "+format(gradient_norm.numpy(), '.5e')+
+                    ", gradient norm divided by the square root of the"+
+                    " number of trainable parameters: "+format(
+                    gradient_norm.numpy()/
+                    self.number_trainable_parameters, '.5e'))
 
         # Evaluates the elapsed time
 
@@ -213,6 +221,10 @@ class ModelCustomTraining:
 
         self.loss_metric = loss_metric
 
+        # Gets the number of trainable parameters of the model
+
+        self.number_trainable_parameters = self.model.count_params()
+
         # Saves the parent path where to save the model
 
         if parent_path is None:
@@ -529,9 +541,12 @@ class ModelCustomTraining:
 
                 print("Iteration group "+integer_toString(i, max_digits)
                 +": loss="+format(loss_value.numpy(), '.5e')+", gradie"+
-                "nt norm: "+format(gradient_value, '.5e')+"\nThis cycl"+
-                "e took "+str(time.time()-self.initial_time)+" to comp"+
-                "lete\n")
+                "nt norm: "+format(gradient_value, '.5e')+", gradient "+
+                "norm divided by the square root of the number of trai"+
+                "nable parameters: "+format(gradient_value/
+                self.number_trainable_parameters, '.5e')+"\nThis cycle"+
+                " took "+str(time.time()-self.initial_time)+" to compl"+
+                "ete\n")
 
         # Otherwise, just call the training function
 

{davout-0.1.1.dev165 → davout-0.1.1.dev167/source/Davout.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: Davout
-Version: 0.1.1.dev165
+Version: 0.1.1.dev167
 Summary: Scientific computing utilities combining FEM, ANN and tooling
 Author: Matheus Janczkowski
 Author-email: Matheus Janczkowski <matheusj2009@hotmail.com>