Save all model layers as trainable by default

src/model/custom_callback.py

@@ -104,8 +104,16 @@ def _save_model(self, subdir: str):
             os.makedirs(outputdir, exist_ok=True)
             model_path = os.path.join(outputdir, "model.keras")
 
-            # Save model
-            self.model.save(model_path)
+            # Create a copy of the model
+            unfrozen_model = tf.keras.models.clone_model(self.model)
+            unfrozen_model.set_weights(self.model.get_weights())
+
+            # Unfreeze all layers in the copied model
+            for layer in unfrozen_model.layers:
+                layer.trainable = True
+
+            # Save the unfrozen model
+            unfrozen_model.save(model_path)
 
             # Save additional files
             if self.charlist:

src/setup/arg_parser.py

@@ -319,6 +319,9 @@ def arg_future_warning(args):
         logger.warning("Argument will lose support in May 2024: "
                        "--config_file_output. The configuration will be saved "
                        "to output/config.json by default.")
+    if args.thaw:
+        logging.warning("Argument will lose support in May 2024: --thaw. "
+                        "Models are saved with all layers thawed by default.")
 
 
 def get_args():