This commit is contained in:
iperov 2021-07-17 22:04:34 +04:00
parent 336e0ca944
commit 9ad9728b40

View File

@ -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 ()