fix
This commit is contained in:
@@ -16,7 +16,7 @@ from core.interact import interact as io
|
|||||||
|
|
||||||
def main(model_class_name, saved_models_path):
|
def main(model_class_name, saved_models_path):
|
||||||
model = models.import_model(model_class_name)(
|
model = models.import_model(model_class_name)(
|
||||||
is_training=False,
|
is_exporting=True,
|
||||||
saved_models_path=saved_models_path,
|
saved_models_path=saved_models_path,
|
||||||
cpu_only=True)
|
cpu_only=True)
|
||||||
model.export_dfm ()
|
model.export_dfm ()
|
||||||
|
|||||||
Reference in New Issue
Block a user