From 724862534053c277efaedaa7a7fa3ef86c963a3c Mon Sep 17 00:00:00 2001
From: Kevin Escobar Rodriguez <kevin.rodriguez@neuralmagic.com>
Date: Wed, 31 Mar 2021 15:26:59 -0400
Subject: [PATCH] Obtaining proper base model for resnet20 (#129)

---
 integrations/keras/prune_resnet20.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/integrations/keras/prune_resnet20.py b/integrations/keras/prune_resnet20.py
index 89b468d48fd..2dfb12f1ea3 100644
--- a/integrations/keras/prune_resnet20.py
+++ b/integrations/keras/prune_resnet20.py
@@ -77,7 +77,7 @@ def download_model_and_recipe(root_dir: str):
         dataset="cifar_10",
         training_scheme=None,
         optim_name="base",
-        optim_category=None,
+        optim_category="none",
         optim_target=None,
         override_parent_path=model_dir,
     )