add model.save and model.load_model

levimcclenny · levimcclenny · commit b164b8e4a932 · 2021-04-02T16:11:31.000-05:00
diff --git a/tensordiffeq/models.py b/tensordiffeq/models.py
@@ -186,7 +186,6 @@ def save(self, path):
     def load_model(self, path):
         self.u_model = tf.keras.models.load_model(path)
 
-
 # WIP
 # TODO Distributed Discovery Model
 class DiscoveryModel():
