diff --git a/mainscripts/ExportDFM.py b/mainscripts/ExportDFM.py index 2b0f978..cf7d64e 100644 --- a/mainscripts/ExportDFM.py +++ b/mainscripts/ExportDFM.py @@ -16,7 +16,7 @@ from core.interact import interact as io def main(model_class_name, saved_models_path): model = models.import_model(model_class_name)( - is_training=False, + is_exporting=True, saved_models_path=saved_models_path, cpu_only=True) model.export_dfm ()